InvalidArgumentError: cannot compute Equal as input #1(zero-based) was expected to be a int32 tensor

本文解决了在使用TensorFlow进行预测时出现的类型不匹配错误。通过调整tf.where函数中的参数类型,确保了比较操作中所有参与者的类型一致,从而避免了InvalidArgumentError异常。

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

InvalidArgumentError: cannot compute Equal as input #1(zero-based) was expected to be a int32 tensor but is a float tensor [Op: Equal] 错误解决!

在这里插入图片描述

在这里插入图片描述
如高亮黄色处,知道Op: Equal是值Equal判断运算出错,tf.equal()函数比较两个元素需要数据类型相同,而上述程序tf.where(PRED.numpy()<0.5,0,1)结果是0或1为int32,而Y类型为float32无法比较,需要将语句tf.where(PRED.numpy()<0.5,0,1)中<0.5后面的0,1改为0. ,1. 变成浮点型。

在这里插入图片描述

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值