md3356 2022-06-05 15:40 采纳率: 62.5%
浏览 12
已结题

关于pytorch中对三维张量的行替换

我现在有一个4x4x2(batch x row x col)的tensor A,还有一个4x2的tensorB。我想要把A中第i个batch的第i行替换成B的第i行,在不使用for的前提下,如何对三维张量进行修改?我知道要用scatter或者index_fill index_put之类的函数,但搜了一圈发现只有二维的,而且也讲的不是很清楚。。

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 6月13日
    • 创建了问题 6月5日