使用以下的函数进行可视化:
# If you use jupyter, uncomment the line below.
# %matplotlib inline
import matplotlib.pyplot as plt
import torchvision
def visualize_batch_tensor(t1):
img=torchvision.utils.make_grid(t1.long()).cpu().numpy()
plt.imshow(np.transpose(img,(1,2,0)))
plt.show()