pytorch零碎笔记

1 关于tensor

  • tensor.size(0) 返回的是第一维的大小
  • squeeze 增加维度
  • unsqueeze 减少维度

猜你喜欢

转载自www.cnblogs.com/rise0111/p/11528192.html
今日推荐