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()

View Code

2.2 输出显示