pytorch项目:测试代码中correct += (y_pred == y).sum().item()

这篇博客探讨了Python中列表和NumPy数组的区别,并通过示例展示了在比较和操作过程中可能出现的问题。在Case1中,尝试将列表直接进行相等比较导致错误,而在Case2和Case3中,使用NumPy数组和PyTorch张量进行比较则能得到预期结果。这强调了在处理数值计算时使用数组库的重要性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

#测试代码中correct += (y_pred == y).sum().item()
import torch
import numpy as np
correct = 0

#case1:当y_pre和y为列表时
y = [1,5,7,8]
y_pred = [2,4,7,9]

correct += (y_pred == y).sum().item() #报错:AttributeError: ‘bool’ object has no attribute ‘sum’

#case2:当y_pre和y为np
y = np.array([3,6,8])
y_pred = np.array([5,6,9])
correct += (y_pred == y).sum().item()
print(correct) # 1

#case3 当为tensor时
y= torch.tensor([5,8,3])
y_pred = torch.tensor([5,8,4])
correct=0
correct += (y_pred == y).sum().item()
print(correct) #2

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值