Browse Source

Slice docstring changes

tags/v0.6.0-beta
hesham 5 years ago
parent
commit
c6c6fbed38
1 changed files with 20 additions and 18 deletions
  1. +20
    -18
      mindspore/dataset/transforms/c_transforms.py

+ 20
- 18
mindspore/dataset/transforms/c_transforms.py View File

@@ -75,28 +75,30 @@ class Slice(cde.SliceOp):
Slice operation to extract a tensor out using the given n slices.

The functionality of Slice is similar to NumPy indexing feature.
(Currently only rank 1 Tensors are supported)
(Currently only rank-1 tensors are supported).

Args:
*slices(Variable length argument list): Maximum `n` number of arguments to slice a tensor of rank `n`.
*slices(Variable length argument list, supported types are, int, list(int), slice, None or Ellipses):
Maximum `n` number of arguments to slice a tensor of rank `n`.
One object in slices can be one of:
1. int: slice this index only. Negative index is supported.
2. slice object: slice the generated indices from the slice object. Similar to `start:stop:step`.
3. None: slice the whole dimension. Similar to `:` in python indexing.
4. Ellipses ...: slice all dimensions between the two slices.
1. :py:obj:`int`: Slice this index only. Negative index is supported.
2. :py:obj:`list(int)`: Slice these indices ion the list only. Negative indices are supdeported.
3. :py:obj:`slice`: Slice the generated indices from the slice object. Similar to `start:stop:step`.
4. :py:obj:`None`: Slice the whole dimension. Similar to `:` in python indexing.
5. :py:obj:`Ellipses`: Slice all dimensions between the two slices. Similar to `...` in python indexing.

Examples:
>>> # Data before
>>> # | col |
>>> # +---------+
>>> # | [1,2,3] |
>>> # +---------|
>>> data = data.map(operations=Slice(slice(1,3))) # slice indices 1 and 2 only
>>> # Data after
>>> # | col |
>>> # +------------+
>>> # | [1,2] |
>>> # +------------|
>>> # Data before
>>> # | col |
>>> # +---------+
>>> # | [1,2,3] |
>>> # +---------|
>>> data = data.map(operations=Slice(slice(1,3))) # slice indices 1 and 2 only
>>> # Data after
>>> # | col |
>>> # +---------+
>>> # | [2,3] |
>>> # +---------|
"""

@check_slice_op
@@ -167,7 +169,7 @@ class PadEnd(cde.PadEndOp):
Pad input tensor according to `pad_shape`, need to have same rank.

Args:
pad_shape (list of `int`): list on integers representing the shape needed. Dimensions that set to `None` will
pad_shape (list(int)): list on integers representing the shape needed. Dimensions that set to `None` will
not be padded (i.e., original dim will be used). Shorter dimensions will truncate the values.
pad_value (python types (str, bytes, int, float, or bool), optional): value used to pad. Default to 0 or empty
string in case of Tensors of strings.


Loading…
Cancel
Save