ノートの切れ端pytorch

テンソルについて1

  • tensor.size(0)は、最初の次元のサイズを返し
  • 増加した寸法を絞ります
  • unsqueeze寸法を縮小

おすすめ

転載: www.cnblogs.com/rise0111/p/11528192.html