pytorch中的Tensor有幂函数吗?

4zcjmb1e  于 2022-12-04  发布在  其他
关注(0)|答案(1)|浏览(135)

pytorch中是否有numpy.power的等价物?
该函数基本上将第一Tensor中的每个元素提升到第二Tensor中的每个对应元素所表示的幂。

xytpbqjk

xytpbqjk1#

您正在寻找torch.pow
正如@Szymon Maszke所提到的,您还可以使用**运算符:

y = torch.pow(x, y)
# the same as
y = x ** y
# and also the same as
y = x.pow(y)

其中y可以是标量或具有broadcastablex形状的Tensor。

相关问题