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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
林间码客3 分钟前
04 ROC曲线与AUC:从零开始手动计算
大数据·人工智能·算法
codexu6 分钟前
NoteGen 里一条记录如何变成 Markdown
人工智能
澄旭14 分钟前
拆解一个成熟 Skill,看懂 Skill 到底该怎么写
人工智能
王木风14 分钟前
拆解一个 LLM 工程化项目:16 个 Service + Agent 对话循环怎么协同跑流水线
人工智能·agent
沪漂阿龙14 分钟前
《LangChain 系列》Human-in-the-loop:什么时候必须让人工介入?
人工智能·架构·langchain
冬哥聊AI18 分钟前
Loop Engineering 来了:从写 Prompt 到设计 Loop,AI 编程的第四次范式跃迁
人工智能
柒星栈22 分钟前
Codex 不只是更强的代码助手,它开始像代理一样推进开发任务了
人工智能
o_insist30 分钟前
04-从零手写 ReAct 循环:Agent 的心跳是怎么转起来的
人工智能·agent
DayByDay30 分钟前
从“单专家”到“多专家辩论”:多大脑对话实现复盘
人工智能
狗哥哥37 分钟前
知乎回答二次创作转AI 漫画/视频思路分享
人工智能