pow

paddle.pow ( x, y, name=None ) [源代码]

该OP是指数算子,逐元素计算 xy 次幂。

pow - 图1

参数

  • x (Tensor)- 多维 Tensor,数据类型为 float32float64int32int64

  • y (float|int|Tensor)- 如果类型是多维 Tensor,其数据类型应该和 x 相同。

  • name (str, 可选)- 操作的名称(可选,默认值为None)。更多信息请参见 Name

返回

Tensor, 维度和数据类型都和 x 相同。

代码示例

  1. import paddle
  2. x = paddle.to_tensor([1, 2, 3], dtype='float32')
  3. # example 1: y is a float or int
  4. res = paddle.pow(x, 2)
  5. print(res)
  6. # Tensor(shape=[3], dtype=float32, place=CUDAPlace(0), stop_gradient=True,
  7. # [1., 4., 9.])
  8. res = paddle.pow(x, 2.5)
  9. print(res)
  10. # Tensor(shape=[3], dtype=float32, place=CUDAPlace(0), stop_gradient=True,
  11. # [1. , 5.65685415 , 15.58845711])
  12. # example 2: y is a Tensor
  13. y = paddle.to_tensor([2], dtype='float32')
  14. res = paddle.pow(x, y)
  15. print(res)
  16. # Tensor(shape=[3], dtype=float32, place=CUDAPlace(0), stop_gradient=True,
  17. # [1., 4., 9.])