深度学习,tensor打印输出不全,出现省略号的两种解决方案
一、设置打印选项,使用torch.set_printoptions函数可以设置在输出中完整显示张量的所有元素。
import torch
# import numpy as np
# 创建一个较大尺寸的张量
tensor_a = torch.ones(100, 1)
# 设置打印选项,显示所有元素
# torch.set_printoptions(threshold=np.inf)
torch.set_printoptions(threshold=float('inf'))
print(tensor_a)
二、转换为NumPy数组:将张量转换为NumPy数组,然后使用NumPy的打印选项来显示所有元素。
import torch
import numpy as np
# 创建一个较大尺寸的张量
large_tensor = torch.ones(100, 1)
# 转换为NumPy数组并使用NumPy的打印选项显示所有元素
print(np.array(large_tensor))