torch.pow¶
- torch.pow(input, exponent, *, out=None) Tensor ¶
使用
exponent
对input
中的每个元素进行幂运算,并返回一个包含结果的张量。exponent
可以是单个float
数字或与input
元素数量相同的 Tensor。当
exponent
是标量值时,应用的运算是当
exponent
是张量时,应用的运算是当
exponent
是张量时,input
和exponent
的形状必须是 可广播的。示例
>>> a = torch.randn(4) >>> a tensor([ 0.4331, 1.2475, 0.6834, -0.2791]) >>> torch.pow(a, 2) tensor([ 0.1875, 1.5561, 0.4670, 0.0779]) >>> exp = torch.arange(1., 5.) >>> a = torch.arange(1., 5.) >>> a tensor([ 1., 2., 3., 4.]) >>> exp tensor([ 1., 2., 3., 4.]) >>> torch.pow(a, exp) tensor([ 1., 4., 27., 256.])
- torch.pow(self, exponent, *, out=None) Tensor
self
是一个标量float
值,而exponent
是一个张量。返回的张量out
与exponent
形状相同。应用的操作是
示例
>>> exp = torch.arange(1., 5.) >>> base = 2 >>> torch.pow(base, exp) tensor([ 2., 4., 8., 16.])