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

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

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

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

而矩阵的单个数值可赋值:

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
正经教主6 分钟前
【FDE系列】阶段3:Day 72:RAGFlow 平台 — 端到端知识库
人工智能·rag·fde
EatFan9 分钟前
AI Agent 进入工程化下半场:从多智能体编排走向治理、标准化与运行沙箱
人工智能·多智能体·ai agent·开源框架·mcp·agents.md
一木 之林10 分钟前
DDPM扩散模型代码实战:前向加噪造数据、U-Net噪声预测与反向生成手写数字
人工智能·算法·计算机视觉
foenix6618 分钟前
AI 程序化建模:用数学构建一朵蘑菇云
人工智能
IvorySQL19 分钟前
PostgreSQL 日报| relchecks 溢出导致表无法删除(9 月 28 日)
数据库·人工智能·ai·postgresql·区块链
workflower23 分钟前
世界模型的内涵与发展
人工智能·机器人·云计算·无人机
9i编程25 分钟前
1. 教 AI 上班:带出我的数字同事 —— 把开发习惯交给 Qoder,从零搭脚手架
人工智能·openai·ai编程
架构师那点事儿28 分钟前
将 HuggingFace 自己的英译中模型迁移到 ONNX
人工智能·python·深度学习
智感子28 分钟前
一维时域信号里,AI 和传统方法谁更靠谱
人工智能
xianghongtao011632 分钟前
麦肯锡2026技术趋势01_智能体式软件开发_研究解读
人工智能