在 PyTorch 中，张量默认的打印方式可能只会显示部分数据（尤其是当张量很大时），为了查看整个张量的数据，你可以使用 torch.set_printoptions 函数来调整打印设置。具体来说，你可以通过设置选项 threshold 来控制打印的长度。将 threshold 参数设置为一个非常大的值，或者使用 float('inf') 将取消长度限制，确保打印出整个张量的数据。

以下是如何做到这一点的示例：

import torch  
# 创建一个示例张量  
large_tensor = torch.arange(100)  
# 默认打印方式（可能是截断的）  
print("Default print:\n", large_tensor)  
# 设置打印选项为无穷以打印完整张量  
torch.set_printoptions(threshold=float('inf'))  
# 现在再次打印张量，这次会显示全部数据  
print("\nFull print:\n", large_tensor)  

Key Steps:

• torch.set_printoptions(threshold=float('inf')):
  • threshold 参数控制打印的元素数量，默认为 1000。设置为 float('inf') 可以取消这个限制，从而打印完整的张量。

注意:

• 使用无限的 threshold 有时可能导致输出过长尤其是对于非常大的张量，因此在实际使用中请结合具体需求调整。
• 如果之后需要恢复到默认打印行为，可以再次调用 torch.set_printoptions(threshold=1000)，将 threshold 恢复到初始值。

通过适当设置这些选项，使得观察和调试张量数据更加方便，有助于理解数据的形态和结果验证。