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 输出显示