首页 > 其他 > 详细

Pytorch对tensor求绝对值

时间:2020-05-02 17:51:52      阅读:120      评论:0      收藏:0      [点我收藏+]

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

 

技术分享图片

 

Pytorch对tensor求绝对值

原文:https://www.cnblogs.com/haifwu/p/12818143.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!