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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
小白说大模型1 分钟前
AI Agent 调试实战:从 Prompt 追踪到执行回放的系统化排障方法
人工智能·prompt
科研小刘带你玩学术8 分钟前
【学术干货】CVPR论文解析:扩散模型如何推动生成式人工智能进入新时代?
人工智能·深度学习·计算机视觉·生成式ai·扩散模型
大模型服务器厂商9 分钟前
AI行业调价潮与产业逻辑解析:算力基建与科研服务器的核心价值
人工智能
Wang's Blog10 分钟前
AI Agent白手起家28: LangChain 五种提示词模板实战解析
大数据·人工智能·langchain
又折桃枝换酒钱11 分钟前
CrossLMM:通过双交叉注意力机制从大型多模态模型中解耦长视频序列
人工智能·深度学习·机器学习
码云之上13 分钟前
Prompt Engineering:从提示词文案到 Agent 行为契约
人工智能·架构·前端工程化
小码哥06818 分钟前
2026陪诊小程序与APP开发技术分析
大数据·人工智能·小程序
观远数据27 分钟前
决策闭环的第三公里:从洞察到行动之间,AI能补上什么
大数据·数据库·人工智能
一次旅行30 分钟前
RLHF全链路深度解析:Reward Model数学推导+PPO完整实战,对比GRPO轻量化方案
人工智能·算法·机器学习
HIT_Weston32 分钟前
164、【Agent】【OpenCode】TuiThreadCmd(工厂设计对比)
人工智能·agent·opencode