Печать torch.tensor в Pytorch: ошибка неверного размера

Question

У меня есть torch.Tensor объект, который имеет форму torch.Size([9, 1, 28, 28])).

Я пробовал что-то вроде

for digit in range(10):
  similar_img = create_interpolates(/*something...*/)
  plt.figure(figsize=(10,10))
  plt.imshow(similar_img.detach().numpy())

Но это дает мне: TypeError: Invalid dimensions for image data

Любая помощь?

Answer 1

Пожалуйста, попробуйте следующий фрагмент кода.

for digit in range(10):
    similar_img = create_interpolates(/*something...*/)
    plt.figure(figsize=(10,10))
    plt.imshow(similar_img[digit].numpy().squeeze())

Спасибо