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