由于有时 tensor 的数据量较大,print 输出 tensor 时会将中间部分的数据用省略号代替(如下图),以致于无法看到全部的参数大小。
在文件开头部分加下面内容即可
import torch torch.set_printoptions(threshold=np.inf)

由于有时 tensor 的数据量较大,print 输出 tensor 时会将中间部分的数据用省略号代替(如下图),以致于无法看到全部的参数大小。
在文件开头部分加下面内容即可
import torch torch.set_printoptions(threshold=np.inf)