forked from mindspore-Ecosystem/mindspore
fix cpu ops example
This commit is contained in:
parent
0459e53040
commit
de025bd228
|
@ -3671,7 +3671,7 @@ class TensorScatterUpdate(PrimitiveWithInfer):
|
||||||
ValueError: If the value of `input_x` are not match with input `indices`.
|
ValueError: If the value of `input_x` are not match with input `indices`.
|
||||||
|
|
||||||
Supported Platforms:
|
Supported Platforms:
|
||||||
``Ascend`` ``GPU``
|
``Ascend`` ``GPU`` ``CPU``
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
>>> input_x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32)
|
>>> input_x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32)
|
||||||
|
@ -4656,7 +4656,7 @@ class SpaceToDepth(PrimitiveWithInfer):
|
||||||
ValueError: If length of shape of `x` is not equal to 4.
|
ValueError: If length of shape of `x` is not equal to 4.
|
||||||
|
|
||||||
Supported Platforms:
|
Supported Platforms:
|
||||||
``Ascend`` ``GPU``
|
``Ascend`` ``GPU`` ``CPU``
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
>>> x = Tensor(np.random.rand(1,3,2,2), mindspore.float32)
|
>>> x = Tensor(np.random.rand(1,3,2,2), mindspore.float32)
|
||||||
|
@ -4725,7 +4725,7 @@ class DepthToSpace(PrimitiveWithInfer):
|
||||||
ValueError: If length of shape of `x` is not equal to 4.
|
ValueError: If length of shape of `x` is not equal to 4.
|
||||||
|
|
||||||
Supported Platforms:
|
Supported Platforms:
|
||||||
``Ascend`` ``GPU``
|
``Ascend`` ``GPU`` ``CPU``
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
>>> x = Tensor(np.random.rand(1, 12, 1, 1), mindspore.float32)
|
>>> x = Tensor(np.random.rand(1, 12, 1, 1), mindspore.float32)
|
||||||
|
|
|
@ -4167,11 +4167,11 @@ class Atanh(PrimitiveWithInfer):
|
||||||
``Ascend`` ``CPU``
|
``Ascend`` ``CPU``
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
>>> x = Tensor(np.array([1.047, 0.785]), mindspore.float32)
|
>>> x = Tensor(np.array([0, -0.5]), mindspore.float32)
|
||||||
>>> atanh = ops.Atanh()
|
>>> atanh = ops.Atanh()
|
||||||
>>> output = atanh(x)
|
>>> output = atanh(x)
|
||||||
>>> print(output)
|
>>> print(output)
|
||||||
[1.8869909 1.058268 ]
|
[0. -0.54930614]
|
||||||
"""
|
"""
|
||||||
|
|
||||||
@prim_attr_register
|
@prim_attr_register
|
||||||
|
@ -4526,7 +4526,7 @@ class Eps(PrimitiveWithInfer):
|
||||||
Tensor, has the same type and shape as `x`, but filled with `x` dtype minimum val.
|
Tensor, has the same type and shape as `x`, but filled with `x` dtype minimum val.
|
||||||
|
|
||||||
Supported Platforms:
|
Supported Platforms:
|
||||||
``Ascend`` ``GPU``
|
``Ascend`` ``GPU`` ``CPU``
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
>>> x = Tensor([4, 1, 2, 3], mindspore.float32)
|
>>> x = Tensor([4, 1, 2, 3], mindspore.float32)
|
||||||
|
|
Loading…
Reference in New Issue