慢跑意味着移动一个向量,使其均值为零。我想将Tensor的每一行居中,例如:
A = |1 , 1| A_center = |0, 0| |0 , 4| |-2,2|
为此,我使用这种方法:
a_center = a - a.mean(dim=1).unsqueeze(1)
我想知道在PyTorch中是否有内置的方法来使Tensor居中。因为我知道有用于长度归一化的内置函数,所以我假设可能也有用于Tensor居中的函数,但我无法找到它。
k0pti3hp1#
到目前为止,Pytorch中还没有内置函数。使用这个:
x_centered = x - x.mean(dim=1, keepdim=True)
1条答案
按热度按时间k0pti3hp1#
到目前为止,Pytorch中还没有内置函数。使用这个: