如何解决Python中'int' object has no attribute 'item'错误？

错误原因很明确：data_sample[1]是Python原生的int类型，而.item()是PyTorch Tensor对象的专属方法，普通整数没有这个属性，因此触发AttributeError。

直接修改两处即可解决问题：

1. 去掉data_sample[1].item()中的.item()，直接将整数转为字符串；
2. 原代码用print调用show_data会输出None（因为函数无返回值），直接调用函数后加上plt.show()才能正常显示图像。

修正后的完整代码：

import matplotlib.pyplot as plt  # 确保已导入matplotlib库

def show_data(data_sample):
    plt.imshow(data_sample[0].numpy().reshape(28, 28), cmap='gray')
    plt.title('y = ' + str(data_sample[1]))  # 移除.item()方法

# 显示第一张图像和标签
show_data(train_dataset[0])
plt.show()  # 显式调用show()展示图像窗口

如果你的数据集标签原本应为Tensor类型，那可能是数据集加载时的设置问题，但根据当前错误信息，标签已经是普通整数，上述修改完全适用。

内容的提问来源于stack exchange，提问作者junaid badshah