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

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

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

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

而矩阵的单个数值可赋值

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

复制代码
tp[tp > 0]=iou
iou_target[idx] = tp
相关推荐
深圳市快瞳科技有限公司几秒前
低功耗设计:如何让AI宠物摄像头实现长达数月的续航?
人工智能
TENSORTEC腾视科技5 分钟前
超低功耗 性能卓越|腾视科技重磅推出TS-SG-SM9系列AI算力模组,引领边缘智能计算新篇章
人工智能·ai·算力模组·模组·ai算力·ai算力模组·ai模组
weixin_408099675 分钟前
医疗票据OCR API vs 在线工具:医院和企业该怎么选?
人工智能·ocr·石榴智能·ocr选型·医疗票据ocr·医疗ocr api·医疗数字化
沪漂阿龙5 分钟前
AI大模型面试题:决策树是什么?ID3、C4.5、CART、信息增益、剪枝一文讲透
人工智能·决策树·剪枝
deephub8 分钟前
Feature Engineering 实战:Pandas + Scikit-learn的机器学习特征工程的完整代码示例
人工智能·python·机器学习·pandas·scikit-learn
自动化测试行业观察9 分钟前
手机银行用户体验测评解决方案
人工智能·智能手机·ux
元拓数智10 分钟前
AI Agent 时代,企业数据治理底座如何支撑智能应用的安全与效率
大数据·人工智能·安全·数据治理·nl2sql·自然语言查询
节点云科10 分钟前
谷歌 Gemini Omni 深度解析:原生视频模型的技术突破与行业影响
人工智能·音视频
奇树谦11 分钟前
2026主流AI Generator全解析:横向对比\+场景化选型指南
人工智能
iDao技术魔方11 分钟前
whisper.cpp 深度解析:从边缘设备到实时语音识别
人工智能·whisper·语音识别