add InplaceAddD and InplaceSubD vm ops

pull/2214/head
liuwenhao4 5 years ago
parent 7038df8b99
commit a743f5b109

@ -777,19 +777,18 @@ class InplaceAdd(PrimitiveWithInfer):
Adds v into specified rows of x. Computes y = x; y[i,] += v. Adds v into specified rows of x. Computes y = x; y[i,] += v.
Args: Args:
- **indices** (Union[int, tuple]) - Indices into the left-most dimension of x, and determines which rows of x indices (Union[int, tuple]): Indices into the left-most dimension of x, and determines which rows of x
to add with v. It is a int or tuple, whose value is in [0, the first dimension size of x). to add with v. It is a int or tuple, whose value is in [0, the first dimension size of x).
Inputs: Inputs:
- **input_x** (Tensor) - The first input is a tensor whose data type is number. - **input_x** (Tensor) - The first input is a tensor whose data type is number.
- **input_v** (Tensor) - The second input is a tensor who has the same dimension sizes as x except - **input_v** (Tensor) - The second input is a tensor who has the same dimension sizes as x except
the first dimension, which must be the same as indices's size. the first dimension, which must be the same as indices's size.
Outputs: Outputs:
Tensor, has the same shape and dtype as input. Tensor, has the same shape and dtype as input.
Examples: Examples:
>>> indices = [0, 1] >>> indices = [0, 1]
>>> input_x = Tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32) >>> input_x = Tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32)
>>> input_v = Tensor(np.array([[0.5, 1.0], [1.0, 1.5]]), mindspore.float32) >>> input_v = Tensor(np.array([[0.5, 1.0], [1.0, 1.5]]), mindspore.float32)
@ -837,13 +836,13 @@ class InplaceSub(PrimitiveWithInfer):
Subtracts v into specified rows of x. Computes y = x; y[i, :] -= v; return y. Subtracts v into specified rows of x. Computes y = x; y[i, :] -= v; return y.
Args: Args:
- **indices** (Union[int, tuple]) - Indices into the left-most dimension of x, and determines which rows of x indices (Union[int, tuple]): Indices into the left-most dimension of x, and determines which rows of x
to sub with v. It is a int or tuple, whose value is in [0, the first dimension size of x). to sub with v. It is a int or tuple, whose value is in [0, the first dimension size of x).
Inputs: Inputs:
- **input_x** (Tensor) - The first input is a tensor whose data type is number. - **input_x** (Tensor) - The first input is a tensor whose data type is number.
- **input_v** (Tensor) - The second input is a tensor who has the same dimension sizes as x except - **input_v** (Tensor) - The second input is a tensor who has the same dimension sizes as x except
the first dimension, which must be the same as indices's size. the first dimension, which must be the same as indices's size.
Outputs: Outputs:
Tensor, has the same shape and dtype as input. Tensor, has the same shape and dtype as input.

Loading…
Cancel
Save