Нанесение факела. Тензор в изображение

Question

same_digit_imgs = torch.empty(0, 1, 28, 28)
for digit in range(10):
  similar_img = create_interpolates(similar_pairs[digit][0], similar_pairs[digit][1], encoder, decoder)
  print(type(similar_img), similar_img.shape)
  similar_img = similar_img.view(similar_img.shape[1], similar_img.shape[2], similar_img.shape[0])
  plt.imshow(similar_img)
  plt.show()

Я пытаюсь построить тензор similar_img в pytorch.Я новичок в Pytorch, и я не уверен, как построить изображение.Тип и форма <class 'torch.Tensor'> torch.Size([9, 1, 28, 28])

, и я получаю сообщение об ошибке: RuntimeError: shape '[1, 28, 9]' is invalid for input of size 7056, когда я сделал similar_img.view(similar_img.shape[1], similar_img.shape[2], similar_img.shape[0]).

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

Answer 1

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

for digit in range(10):
    similar_img = create_interpolates(similar_pairs[digit][0], similar_pairs[digit] 
    [1], encoder, decoder)
    print(type(similar_img), similar_img.shape)
    plt.imshow(similar_img[digit].numpy().squeeze())
    plt.show()

Спасибо