mindspore.dataset.text.SlidingWindow

class mindspore.dataset.text.SlidingWindow(width, axis=0)[源代码]

在输入数据的某个维度上进行滑窗切分处理,当前仅支持处理1-D的Tensor。

参数:
  • width (int) - 窗口的宽度,它必须是整数并且大于零。

  • axis (int, 可选) - 计算滑动窗口的轴。默认值:0。

异常:
  • TypeError - 参数 width 的类型不为int。

  • ValueError - 参数 width 的值不为正数。

  • TypeError - 参数 axis 的类型不为int。

支持平台:

CPU

样例:

>>> import mindspore.dataset as ds
>>> dataset = ds.NumpySlicesDataset(data=[[1, 2, 3, 4, 5]], column_names="col1")
>>> # Data before
>>> # |     col1     |
>>> # +--------------+
>>> # | [[1, 2, 3, 4, 5]] |
>>> # +--------------+
>>> dataset = dataset.map(operations=text.SlidingWindow(3, 0))
>>> # Data after
>>> # |     col1     |
>>> # +--------------+
>>> # |  [[1, 2, 3], |
>>> # |   [2, 3, 4], |
>>> # |   [3, 4, 5]] |
>>> # +--------------+