mindspore.ops.cond

View Source On AtomGit
mindspore.ops.cond(A, p=None)[source]

Return the matrix norm or vector norm of a given tensor.

p is the calculation mode of norm. The following norm modes are supported.

Note

Currently, complex numbers are not supported.

Parameters:

  • A (Tensor) – The input tensor which has zero or more batch dimensions.

  • p (Union[int, float, inf, -inf, 'fro', 'nuc'], optional) – Norm's mode. Refer to the table above for behavior. Default None.

Returns:

Tensor

Supported Platforms:

GPU CPU

Examples

>>> import mindspore
>>> x = mindspore.tensor([[1.0, 0.0, -1.0], [0.0, 1.0, 0.0], [1.0, 0.0, 1.0]])
>>> print(mindspore.ops.cond(x))
1.4142
>>> print(mindspore.ops.cond(x, 'fro'))
3.1622777