Skip to content

Commit

Permalink
[Fix] tensor_split docstring of supported data type
Browse files Browse the repository at this point in the history
  • Loading branch information
megemini committed Nov 14, 2023
1 parent 758192d commit fc8fdbd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/paddle/tensor/manipulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -2199,7 +2199,7 @@ def tensor_split(x, indices_or_sections, axis=0, name=None):
Split the input tensor into multiple sub-Tensors along ``axis``, allowing not being of equal size.
Args:
x (Tensor): A Tensor whose dimension must be greater than 0. The data type is bool, bfloat16, float16, float32, float64, uint8, int8, int16, int32, complex64, complex128 or int64.
x (Tensor): A Tensor whose dimension must be greater than 0. The data type is bool, bfloat16, float16, float32, float64, uint8, int8, int32 or int64.
indices_or_sections (int|list|tuple): If ``indices_or_sections`` is an int ``n``, ``x`` is split into ``n`` sections along ``axis``.
If ``x`` is divisible by ``n``, each section will be ``x.shape[axis] / n``. If ``x`` is not divisible by ``n``, the first
``int(x.shape[axis] % n)`` sections will have size ``int(x.shape[axis] / n) + 1``, and the rest will be ``int(x.shape[axis] / n).
Expand Down

0 comments on commit fc8fdbd

Please sign in to comment.