Python PyTorch pow用法及代碼示例

本文簡要介紹python語言中 torch.pow 的用法。

用法:

torch.pow(input, exponent, *, out=None) → Tensor

參數：

• input(Tensor) -輸入張量。

• exponent(float或者張量) - index 值

關鍵字參數：

out(Tensor,可選的) -輸出張量。

參數：

• self(float) -冪運算的標量基值

• exponent(Tensor) - index 張量

關鍵字參數：

out(Tensor,可選的) -輸出張量。

使用 exponent 獲取 input 中每個元素的冪並返回帶有結果的張量。

exponent 可以是單個 float 編號，也可以是具有與 input 相同數量的元素的 Tensor。

當exponent 為標量值時，應用的操作為：

\text{out}_i = x_i ^ \text{exponent}

當exponent 是張量時，應用的操作是：

\text{out}_i = x_i ^ {\text{exponent}_i}

當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 的形狀相同

應用的操作是：

\text{out}_i = \text{self} ^ {\text{exponent}_i}

例子：

>>> exp = torch.arange(1., 5.)
>>> base = 2
>>> torch.pow(base, exp)
tensor([  2.,   4.,   8.,  16.])