docstring bug fix

pull/12276/head
wangrao 4 years ago
parent 3f81802498
commit 9d63b443e3

@ -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

@ -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