Pytorch對tensor求絕對值

1. torch.abs(self: Tensor, *, out: Optional[Tensor]=None)

pytorch中對tensor求絕對值，使用的方法是torch.abs()，跟numpy中求絕對值的方法類似，numpy中使用numpy.abs()。

參數說明：

self：傳入的參數

out：輸出，默認即可，不用設定。

即只要傳入要求絕對值的tensor即可。

2. 例子

2.1 Code

mport torch


def res_abs():
    data = torch.tensor([[[[1, -1],
                           [-2, -3]]]])
    print(data.shape)
    print("data: ", data)
    data = torch.abs(data)
    print("data_abs: ", data)


if __name__ == "__main__":
    res_abs()

2.2 輸出顯示