pytorch的二次索引矩阵无法赋值问题

最近在研究中发现torch一个问题,即torch的二次索引的矩阵无法赋值。

具体来说,给定相同的初始常数矩阵aiou_target 矩阵, 以及另一iou矩阵,直接赋值是没问题的。

然而,当对iou_target矩阵进行二次索引时,即idx矩阵和tp矩阵,可以看到赋值并未起到作用:

而矩阵的单个数值可赋值

根据以上观测,在二次索引赋值时,需通过中间变量tp进行。上述过程可以实现为:

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
AI科技星14 小时前
光的几何起源:从螺旋时空到量子现象的完全统一
开发语言·人工智能·线性代数·算法·机器学习
小程故事多_8014 小时前
打破传统桎梏,LLM 让智能运维实现从 “自动化” 到 “自进化”
运维·人工智能·自动化·aigc
星爷AG I14 小时前
9-9 数量与密度(AGI基础理论)
人工智能·agi
Tiaoxiaobai14 小时前
如何实现亚细胞定位
人工智能·笔记
cc_beolus14 小时前
昇腾AI入门
人工智能
AI即插即用14 小时前
即插即用系列 | CVPR 2025 SegMAN: Mamba与局部注意力强强联合,多尺度上下文注意力的新SOTA
图像处理·人工智能·深度学习·目标检测·计算机视觉·视觉检测
q_354888515314 小时前
机器学习:Python地铁人流量数据分析与预测系统 基于python地铁数据分析系统+可视化 时间序列预测算法 ✅
大数据·人工智能·python·算法·机器学习·信息可视化·数据分析
房产中介行业研习社14 小时前
2026年1月房产中介管理系统评测
大数据·人工智能
莱昂纳多迪卡普利奥14 小时前
LLM学习指南(五)——大语言模型(LLM)
人工智能·语言模型·自然语言处理
方见华Richard14 小时前
认知几何学:思维如何弯曲意义空间V0.3
人工智能·经验分享·交互·原型模式·空间计算