Блог

Я работаю над GAN, и я хочу визуализировать сформированное изображение.

Для этого я пытался

 def show_images(image_tensor, num_images=9, size=(1, 28, 28)):
    image_unflat = image_tensor.detach().cpu.view(-1, *size)
    image_grid = make_grid(image_unflat[:num_images], nrow=3)
    plt.imshow(image_grid.permute(1, 2, 0).squeeze())
    plt.show()
 

но когда я пытаюсь show_image(some_tensor) , я получаю сообщение об ошибке как

 image_unflat = image_tensor.detach().cpu.view(-1, *size)
AttributeError: 'builtin_function_or_method' object has no attribute 'view'
 

Здесь размер some_tensor равен N x 784.

Вам нужно позвонить cpu() перед трансляцией с view помощью .

 image_unflat = image_tensor.detach().cpu().view(-1, *size)