在 pytorch 中将张量居中

Question

慢跑意味着移动向量，使其均值为零。

我想将张量的每一行居中，例如：

A = |1 , 1|    A_center = |0, 0|
    |0 , 4|               |-2,2|

为此，我使用这种方法：

a_center  = a - a.mean(dim=1).unsqueeze(1)

我想知道 PyTorch 中是否有任何内置方法可以将张量居中。因为我知道有用于长度归一化的内置函数，所以我认为也可能有一个用于张量居中的函数，但我找不到它。

Answer 1

到目前为止，Pytorch 中还没有内置函数。使用这个：

x_centered = x - x.mean(dim=1, keepdim=True)