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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
盛世宏博北京几秒前
学校图书馆自动化恒温恒湿控制系统技术方案
网络·数据库·人工智能
神州问学4 分钟前
每周技术加速器:UltraRAG:突破传统RAG架构的创新与实践
人工智能
GitCode官方14 分钟前
YOLO11 与 Wan2.2‑I2V‑A14B 正式上线 AtomGit AI:开启视觉感知与动态生成新纪元!
人工智能·计算机视觉·目标跟踪·开源·atomgit
deephub17 分钟前
机器学习时间特征处理:循环编码(Cyclical Encoding)与其在预测模型中的应用
人工智能·python·机器学习·特征工程·时间序列
Gofarlic_oms119 分钟前
集中式 vs 分布式许可:跨地域企业的管控架构选择
大数据·运维·人工智能·分布式·架构·数据挖掘·需求分析
机器学习之心21 分钟前
科研绘图 | PSO-LSTM粒子群优化长短期记忆神经网络模型结构图
人工智能·神经网络·lstm·pso-lstm
meizisay24 分钟前
亿可达_自动发邮件攻略
人工智能·经验分享·低代码·职场和发展·自动化
褪色的博客27 分钟前
强化学习入门:核心概念与数学基础详解
人工智能
遥感学习森29 分钟前
滑雪场分布API及滑雪预报服务API
大数据·人工智能·天气api
code bean31 分钟前
【AI】RAG智能问答的三层优化策略
大数据·人工智能·rag