当前位置： 移动技术网 > IT编程>脚本编程>Python > Pytorch对tensor求绝对值

Pytorch对tensor求绝对值

2020年05月03日  | 移动技术网IT编程  | 我要评论

九阴真经孔雀石,顾城杀妻,格瑞儿童装

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

 

 

您可能感兴趣的文章:

如对本文有疑问，请在下面进行留言讨论，广大热心网友会与你互动！！ 点击进行留言回复

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网