TensorFlow加轴的bug

在编程过程中遇到一个涉及 TensorFlow 的 bug,错误地尝试在同一步骤中进行按轴取数和增加维度操作。正确做法是将这两步分开,例如:count_value=rules[:,1][tf.newaxis]用于正确地按轴取数并新增一维,而错误用法如 count_value=rules[tf.newaxis,:,1] 会导致随机错误。理解这种区别对于避免类似问题至关重要。

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

写代码的过程中遇到一个特别的bug,会产生随机的错误,记录一下,
当我们想同时完成按轴取数并且新加一维的时候,要放到两步里面做,而不是同一步里面完成

正确用法用法

count_value = rules[:, 1][tf.newaxis] # 1, M

错误用法

count_value = rules[tf.newaxis,:,1]

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值