如何输出全部的Tensor,而不是使用省略号掠过

最近在模型的debug时经常需要输出模型每一步的Tensor,但是默认是使用省略号掠过大部分Tensor,因此我们没办法很好的观察到全部的Tensor。

我们可以使用以下函数使每次输出都会展示全部的Tensor:

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

你可能感兴趣的:(kill_bug,pytorch)