Получение тензора формы (32) вместо одного значения в качестве метки ⇐ Python

[Anonymous]

когда я пытаюсь напечатать/plt.tile метку изображения, он печатает тензор (32,) вместо одного значения метки. Когда я пытался распечатать до изменения размера, печать шла нормально.
Код:

Код: Выделить всё

####

# importing libraries

import tensorflow as tf
tf.random.set_seed(1337)
import tensorflow_datasets as tfds

# Splitting dataset

train_ds, validation_ds, test_ds = tfds.load("cats_vs_dogs",
# Reserve 10% for validation and 10% for test
split=["train[:10%]",
"train[40%:50%]",
"train[50%:60%]"],
as_supervised=True, # Include labels
)

# resizing test dataset

size = (150, 150)
test_ds = test_ds.map(lambda x, y: (tf.image.resize(x, size), y))

# Prefetch data for optimized loading.

train_ds      = train_ds.cache().batch(batch_size).prefetch(buffer_size=10)
validation_ds = validation_ds.cache().batch(batch_size).prefetch(buffer_size=10)
test_ds       = test_ds.cache().batch(batch_size).prefetch(buffer_size=10)

# plotting the dataset

for i, (imagexd, labelxd) in enumerate(test_ds.take(9)):
yea_image = imagexd[0].numpy().astype("int32")
axd = plt.subplot(3,3, i+1)

plt.imshow(yea_image)
plt.axis("off")
#plt.title(f"Label is {int(labelxd)}")

####
plt.title((labelxd))
print(labelxd)
break
####

####
# plt.title(y_true[i])
####

####
# print(i)
####

####

Снимок экрана:
Изображение

Подробнее здесь: https://stackoverflow.com/questions/798 ... as-a-label