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 輸出顯示

轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/150693.html
標籤:Python
上一篇:1.JVM基礎知識
