Lar*_*ial 5 python operator-overloading pytorch
假设x
是 Pytorch 中的一个张量。一个人可以写:
x_lowerthanzero = x.lt(0)
Run Code Online (Sandbox Code Playgroud)
或者:
x_lowerthanzero = (x<0)
Run Code Online (Sandbox Code Playgroud)
看似完全相同的结果。许多其他的操作都Pytorch内置等效:x.gt(0)
对(x>0)
,x.neg()
对-x
,x.mul()
等等。
是否有充分的理由使用一种形式而不是另一种形式?
通常没有理由使用其中一个,它们主要是为了方便:其中许多方法确实有一个out
参数,它可以让您指定一个张量来保存结果,但您也可以这样做使用运算符而不是方法。