python - 打印张量的所有内容

Question

我遇到了 this PyTorch 教程(在 neural_networks_tutorial.py 中)，他们在其中构建了一个简单的神经网络并运行了推理。为了调试目的，我想打印整个输入张量的内容。当我尝试打印张量时得到的是这样的而不是整个张量:

我看到了一个类似的 link对于 numpy，但不确定什么适用于 PyTorch。我可以将它转换为 numpy 并可以查看它，但我想避免额外的开销。有没有办法打印整个张量？

Answer 1

要避免截断并控制打印多少张量数据，请使用与 numpy 的 numpy.set_printoptions(threshold=10_000) 相同的 API。

例子:

x = torch.rand(1000, 2, 2)
print(x) # prints the truncated tensor
torch.set_printoptions(threshold=10_000)
print(x) # prints the whole tensor

如果您的张量非常大，请将threshold 值调整为更高的值。

另一种选择是:

torch.set_printoptions(profile="full")
print(x) # prints the whole tensor
torch.set_printoptions(profile="default") # reset
print(x) # prints the truncated tensor

所有可用的 set_printoptions 参数均已记录 here .