pytorch输出tensor张量时显示省略号的问题

问题描述：

　　由于tensor的数据量过大，再用print输出tensor数据时会将中间的部分用省略号代替，无法看到全部的tensor元素。

 

 

 

解决方案：

　　在程序开头加入下面的代码即可

import torch
import numpy as np
torch.set_printoptions(threshold=np.inf)