Browse Source

fix bugs

tags/v1.1.0
lihongkang 5 years ago
parent
commit
639a085d93
3 changed files with 4 additions and 0 deletions
  1. +2
    -0
      mindspore/ops/operations/array_ops.py
  2. +1
    -0
      mindspore/ops/operations/math_ops.py
  3. +1
    -0
      mindspore/ops/operations/nn_ops.py

+ 2
- 0
mindspore/ops/operations/array_ops.py View File

@@ -826,6 +826,7 @@ class Rank(PrimitiveWithInfer):
>>> input_tensor = Tensor(np.array([[2, 2], [2, 2]]), mindspore.float32)
>>> rank = P.Rank()
>>> rank(input_tensor)
2
"""

@prim_attr_register
@@ -896,6 +897,7 @@ class Size(PrimitiveWithInfer):
>>> input_tensor = Tensor(np.array([[2, 2], [2, 2]]), mindspore.float32)
>>> size = P.Size()
>>> output = size(input_tensor)
4
"""

@prim_attr_register


+ 1
- 0
mindspore/ops/operations/math_ops.py View File

@@ -2938,6 +2938,7 @@ class Sin(PrimitiveWithInfer):
>>> sin = P.Sin()
>>> input_x = Tensor(np.array([0.62, 0.28, 0.43, 0.62]), mindspore.float32)
>>> output = sin(input_x)
[0.5810352 0.27635565 0.41687083 0.5810352]
"""

@prim_attr_register


+ 1
- 0
mindspore/ops/operations/nn_ops.py View File

@@ -1761,6 +1761,7 @@ class SparseSoftmaxCrossEntropyWithLogits(PrimitiveWithInfer):

@prim_attr_register
def __init__(self, is_grad=False):
validator.check_value_type('is_grad', is_grad, [bool], self.name)
self.init_prim_io_names(inputs=['features', 'labels'], outputs=['output'])
self.is_grad = is_grad
self.add_prim_attr('sens', 1.0)


Loading…
Cancel
Save