Plotting a tensor

1

Hello! I have a tensor avg_act['a'] of size (64,2000) and I try

plt.imshow(avg_act['a'][idx1])

and get this error: Invalid shape (128,) for image data
How can I fix this?

2

Hi,
your tensor needs to have at least 2 dimension for imshow. Check the docs for reference.

3

now that I have one dimension what should I use? Is there anything?

4

You could just unsqueeze the data you pass to imshow.

5

I don’t know how. Could you please explain?

6

By using example_tensor.unsqueeze(-1) for instance.

7

thank you! I did that and now the code works without errors but it doesn’t show any plots. what is wrong? I’m using python 3.7 on CPU.

8

Have you forgotten to use plt.show()?

9

Yes! Thank you it works now

1 Like