pytorch输出tensor张量时有省略号的解决方案（将tensor完整输出）

问题描述

由于有时 tensor 的数据量较大，print 输出 tensor 时会将中间部分的数据用省略号代替（如下图），以致于无法看到全部的参数大小。

解决方案

在文件开头部分加下面内容即可

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