pytorch中是否有numpy.power的等价物?该函数基本上将第一Tensor中的每个元素提升到第二Tensor中的每个对应元素所表示的幂。
numpy.power
xytpbqjk1#
您正在寻找torch.pow。正如@Szymon Maszke所提到的,您还可以使用**运算符:
torch.pow
**
y = torch.pow(x, y) # the same as y = x ** y # and also the same as y = x.pow(y)
其中y可以是标量或具有broadcastable到x形状的Tensor。
y
x
1条答案
按热度按时间xytpbqjk1#
您正在寻找
torch.pow
。正如@Szymon Maszke所提到的,您还可以使用
**
运算符:其中
y
可以是标量或具有broadcastable到x
形状的Tensor。