mindspore.dataset.transforms.Slice

class mindspore.dataset.transforms.Slice(*slices)[源代码]

使用给定的slices,对Tensor进行切片操作。功能类似于NumPy的索引(目前只支持1D形状的Tensor)。

参数:
  • slices (Union[int, list[int], slice, None, Ellipsis]) - 指定切片的信息,可以为

      1. int: 沿着第一个维度切片对索引进行切片,支持负索引。

      1. list(int): 沿着第一个维度切片所有索引进行切片,支持负号索引。

      1. slice: 沿着第一个维度对 slice 对象生成的索引进行切片。

      1. None: 切片整个维度,类似于Python索引中的语法 [:]

      1. Ellipsis: 切片整个维度,效果与 None 相同。

异常:
  • TypeError - 参数 slices 类型不为int、list[int]、sliceNoneEllipsis

支持平台:

CPU

样例:

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