numpy 在pytorch中中心Tensor

z18hc3ub  于 12个月前  发布在  其他
关注(0)|答案(1)|浏览(130)

慢跑意味着移动一个向量,使其均值为零。
我想将Tensor的每一行居中,例如:

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

为此,我使用这种方法:

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

我想知道在PyTorch中是否有内置的方法来使Tensor居中。因为我知道有用于长度归一化的内置函数,所以我假设可能也有用于Tensor居中的函数,但我无法找到它。

k0pti3hp

k0pti3hp1#

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

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

相关问题