Browse Source

docstring bug fix

tags/v1.2.0-rc1
wangrao 4 years ago
parent
commit
9d63b443e3
2 changed files with 3 additions and 1 deletions
  1. +1
    -1
      mindspore/numpy/array_ops.py
  2. +2
    -0
      mindspore/numpy/math_ops.py

+ 1
- 1
mindspore/numpy/array_ops.py View File

@@ -561,7 +561,7 @@ def vstack(tup):
"""
Stacks tensors in sequence vertically.
This is equivalent to concatenation along the first axis. 1-D tensors should firstly be reshaped to `(1, N)`,
and then be concatenated along the first axis.
and then be concatenated along the first axis.

Args:
tup (Union[Tensor, tuple, list]): A sequence of 1-D or 2-D tensors. The tensors must have the same shape


+ 2
- 0
mindspore/numpy/math_ops.py View File

@@ -963,11 +963,13 @@ def tensordot(a, b, axes=2):
non-negative integer_like scalar, `N`; if it is such, then the last `N` dimensions of
`a` and the first `N` dimensions of `b` are summed over.
Three common use cases are:

``axes = 0`` : tensor product

``axes = 1`` : tensor dot product

``axes = 2`` : (default) tensor double contraction

When axes is integer_like, the sequence for evaluation will be: first the `-Nth`
axis in `a` and 0th axis in `b`, and the -1th axis in `a` and `Nth` axis in `b` last.
When there is more than one axis to sum over - and they are not the last (first)


Loading…
Cancel
Save