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