方法一:(调用函数来看)
问题描述
由于有时 tensor 的数据量较大,print 输出 tensor 时会将中间部分的数据用省略号代替(如下图),以致于无法看到全部的参数大小。
解决方案
在文件开头部分加下面内容即可
- import torch
- import numpy as np
- torch.set_printoptions(threshold=np.inf)或是torch.set_printoptions(profile="full")
注意的是:直接调试的时候看;输出出来不方便看,倒是也能看
这部分内容参照:pytorch输出tensor张量时有省略号的解决方案(将tensor完整输出)_pytorch显示完整tensor_friedrichor的博客-CSDN博客
方法二:(转numpy来看)
<tensor>.numpy()
把