pow

paddle.sparse. pow ( x, factor, name=None ) [source]

Calculate elementwise pow of x, requiring x to be a SparseCooTensor or SparseCsrTensor.

\[out = x^{factor}\]
Parameters
  • x (Tensor) – The input Sparse Tensor with data type float32, float64.

  • factor (float|int) – factor of pow.

  • name (str, optional) – Name for the operation (optional, default is None). For more information, please refer to Name.

Returns

A Sparse Tensor with the same data type and shape as x .

Examples

>>> import paddle

>>> dense_x = paddle.to_tensor([-2, 0, 3], dtype='float32')
>>> sparse_x = dense_x.to_sparse_coo(1)
>>> out = paddle.sparse.pow(sparse_x, 2)
>>> out
Tensor(shape=[3], dtype=paddle.float32, place=Place(cpu), stop_gradient=True,
    indices=[[0, 2]],
    values=[4., 9.])