.numpy()

[JoshK68]

Hi,

Quick question...Why do we need to call .numpy() here:

Plot an image tensor

import matplotlib.pyplot as plt
plt.imshow(image)
plt.title(class_names[label.numpy()]) # HERE
plt.axis(False);

Trying to understand why it's necessary.

Thank you,
Josh

[gauravreddy08]

Hey @JoshK68,

Here label refers to a Tensor. While indexing a list we generally don't use a tensor. So the code snippet converts it into NumPy datatype using .numpy()

[JoshK68]

Hi @gaurxvreddy,

What do you mean with: 'While indexing a list we cannot you a tensor.' ?

Thank you!
Josh

[mrdbourke]

Plot an image tensor

import matplotlib.pyplot as plt
plt.imshow(image)
plt.title(class_names[label.numpy()]) # HERE
plt.axis(False);

Hey Josh,

Did you try without numpy() and see what happened?

I think you might find an error - matplotlib can't work with TensorFlow tensors as titles.