Pytorch对tensor求绝对值

佚名 4年前 (2020-05-02) Python 1644人围观抢沙发百度已收录

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

pytorch中对tensor求绝对值，使用的方法是torch.abs()，跟numpy中求绝对值的方法类似，numpy中使用numpy.abs()。

参数说明：

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

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