【发布时间】：2020-11-16 12:09:35
【问题描述】：
pytorch 中有 numpy.power 的等价物吗？
该函数基本上将第一个张量中的每个元素提升到第二个张量中每个对应元素所代表的幂。
【问题讨论】：
标签： numpy deep-learning neural-network pytorch
【发布时间】：2020-11-16 12:09:35
【问题描述】：
您正在寻找torch.pow。
【讨论】：
-
torch.pow仅对所有元素应用标量幂。numpy.power在两个张量之间应用逐元素幂。 -
@Light
torch.pow(tensor1, tensor2)按元素工作。您也可以使用tensor1 ** tensor2或tensor1.pow(tensor2)。