torch打印tensor省略的情况

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

猜你喜欢

转载自blog.csdn.net/qq_43263543/article/details/128001713