我正在尝试弄清楚如何在b (batch size), d (depth), h (hight) and w (width)表示为如下的张量上进行移位: b, d, h, w = tensor.size() 因此,我需要找到移位张量和张量本身之间的减法。我正在考虑使用torch.narrow或torch.concat对每一边进行减法(向右、向左、向上和向下),每次我从相同的张量一侧(张量本身一侧)减去,最后我将对每一侧
例如,我有一个N B x H张量列表(即N x B x H张量)和一个N向量列表(即N x B张量)。我要把列表中的每个B x H张量乘以相应的B维张量,从而得到N x H张量。我知道如何使用单个for-loop和PyTorch来实现计算,但是有矢量注入吗?(即不使用for-loop,只使用PyTorch/numpy操作)