技术教程 Pytorch对tensor求绝对值 正文

Pytorch对tensor求绝对值

2022-10-08,,

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

 

 

《Pytorch对tensor求绝对值.doc》

下载本文的Word格式文档，以方便收藏与打印。

Copyright 2023. 昆居客 - IT编程,互联网信息技术文档大全!