切片

切片 - 13

版本

  • 名称: 切片 (GitHub)

  • : main

  • 自版本: 13

  • 功能: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符已从版本 13开始提供。

摘要

在多个轴上生成输入张量的切片。类似于 numpy: https://numpy.com.cn/doc/stable/user/basics.indexing.html?highlight=slice#slicing-and-striding

切片使用 startsendsaxessteps 输入来选择其输入 data 张量的子张量。

对于 [0, ... r-1] 中的每个 i,必须计算一个有效的 starts[i]ends[i]steps[i],其中 r = rank(input),如下所示

如果省略了 axes,则将其设置为 [0, ..., r-1]。如果省略了 steps,则将其设置为长度为 len(starts)[1, ..., 1]

有效值初始化为 start[i] = 0ends[i] = dims[i],其中 dimsinput 的维度,而 steps[i] = 1

通过向 axes 的所有负元素添加 r 来使它们变为非负数,其中 r =rank(input)

通过向 starts[i]ends[i] 中的所有负值添加 dims[axes[i]] 来使它们变为非负数,其中 dimsinput 的维度。然后,start[axes[i]] 是已调整的 starts[i],将其钳制到范围 [0, dims[axes[i]]](对于正向步长)和 [0, dims[axes[i]]-1](对于负向步长)内。

已调整的 ends[i] 的钳制取决于 steps[i] 的符号,并且必须容纳复制 0 到 dims[axes[i]] 个元素,因此对于正向步长,ends[axes[i]] 被钳制到 [0, dims[axes[i]]],而对于负向步长,则被钳制到 [-1, dims[axes[i]]-1]

最后,steps[axes[i]] = steps[i]

对于切片到具有未知大小的维度的末尾,建议在向前切片时传入 INT_MAX,而在向后切片时传入“INT_MIN”。

示例 1

data = [
    [1, 2, 3, 4],
    [5, 6, 7, 8],
]
axes = [0, 1]
starts = [1, 0]
ends = [2, 3]
steps = [1, 2]
result = [
    [5, 7],
]

示例 2

data = [
    [1, 2, 3, 4],
    [5, 6, 7, 8],
]
starts = [0, 1]
ends = [-1, 1000]
result = [
    [2, 3, 4],
]

输入

3 到 5 个输入。

  • 数据 (异构) - T

    要从中提取切片的數據張量。

  • 开始 (异构) - Tind

    axes 中相应轴的起始索引的 1 维张量。

  • 结束 (异构) - Tind

    axes 中相应轴的结束索引(不含)的 1 维张量。

  • (可选,异构) - Tind

    startsends 应用到的轴的 1 维张量。负值表示从后面开始计数维度。接受的范围是 [-r, r-1],其中 r = rank(data)。如果轴重复,则行为未定义。

  • 步长 (可选,异构) - Tind

    axes 中相应轴的切片步长的 1 维张量。负值表示向后切片。“步长”不能为 0。默认为 1。

输出

  • 输出 (异构) - T

    切片后的数据张量。

类型约束

  • T in ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。

  • Tind in ( tensor(int32), tensor(int64) )

    将索引约束为整数类型。

切片 - 11

版本

  • 名称: 切片 (GitHub)

  • : main

  • 版本: 11

  • 功能: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

该操作符的这个版本从版本 11 开始可用

概述

沿着多个轴生成输入张量的切片。类似于 numpy: https://numpy.com.cn/doc/stable/reference/routines.indexing.html 切片使用 startsendsaxessteps 输入来指定轴列表中每个轴的开始和结束维度以及步长,它使用此信息来切片输入 data 张量。如果为任何开始或结束索引传递负值,则表示该维度结束前的元素数量。如果传递给开始或结束的值大于 n(该维度中的元素数量),则表示 n。对于切片到未知大小的维度的末尾,建议在向前切片时传入 INT_MAX,在向后切片时传入‘INT_MIN’。如果为步长传递负值,则表示向后切片。但是,步长值不能为 0。如果省略了 axes,则将其设置为 [0, ..., ndim-1]。如果省略了 steps,则将其设置为 [1, ..., 1],长度为 len(starts) 示例 1:data = [ [1, 2, 3, 4], [5, 6, 7, 8], ] axes = [0, 1] starts = [1, 0] ends = [2, 3] steps = [1, 2] result = [ [5, 7], ] 示例 2:data = [ [1, 2, 3, 4], [5, 6, 7, 8], ] starts = [0, 1] ends = [-1, 1000] result = [ [2, 3, 4], ]

输入

3 到 5 个输入。

  • 数据 (异构) - T

    要从中提取切片的數據張量。

  • 开始 (异构) - Tind

    axes 中相应轴的起始索引的 1 维张量。

  • 结束 (异构) - Tind

    axes 中相应轴的结束索引(不含)的 1 维张量。

  • (可选,异构) - Tind

    1-D 张量,表示 startsends 应用到的轴。负值表示从后向前计数维度。接受范围为 [-r, r-1],其中 r = rank(data)。

  • 步长 (可选,异构) - Tind

    1-D 张量,表示 axes 中对应轴的切片步长。负值表示向后切片。‘steps’ 不能为 0。默认为 1。

输出

  • 输出 (异构) - T

    切片后的数据张量。

类型约束

  • T in ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。

  • Tind in ( tensor(int32), tensor(int64) )

    将索引约束为整数类型。

切片 - 10

版本

  • 名称: 切片 (GitHub)

  • : main

  • 版本: 10

  • 功能: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

该操作符的这个版本从版本 10 开始可用

概述

沿着多个轴生成输入张量的切片。类似于 numpy: https://numpy.com.cn/doc/stable/reference/routines.indexing.html 切片使用 startsendsaxessteps 输入来指定轴列表中每个轴的开始和结束维度以及步长,它使用此信息来切片输入 data 张量。如果为任何开始或结束索引传递负值,则表示该维度结束前的元素数量。如果传递给开始或结束的值大于 n(该维度中的元素数量),则表示 n。对于切片到未知大小的维度的末尾,建议传入 INT_MAX。如果为步长传递负值,则表示向后切片。如果省略了 axes,则将其设置为 [0, ..., ndim-1]。如果省略了 steps,则将其设置为 [1, ..., 1],长度为 len(starts) 示例 1:data = [ [1, 2, 3, 4], [5, 6, 7, 8], ] axes = [0, 1] starts = [1, 0] ends = [2, 3] steps = [1, 2] result = [ [5, 7], ] 示例 2:data = [ [1, 2, 3, 4], [5, 6, 7, 8], ] starts = [0, 1] ends = [-1, 1000] result = [ [2, 3, 4], ]

输入

3 到 5 个输入。

  • 数据 (异构) - T

    要从中提取切片的數據張量。

  • 开始 (异构) - Tind

    axes 中相应轴的起始索引的 1 维张量。

  • 结束 (异构) - Tind

    axes 中相应轴的结束索引(不含)的 1 维张量。

  • (可选,异构) - Tind

    1-D 张量,表示 startsends 应用到的轴。

  • 步长 (可选,异构) - Tind

    1-D 张量,表示 axes 中对应轴的切片步长。默认为 1。

输出

  • 输出 (异构) - T

    切片后的数据张量。

类型约束

  • T in ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。

  • Tind in ( tensor(int32), tensor(int64) )

    将索引约束为整数类型。

切片 - 1

版本

  • 名称: 切片 (GitHub)

  • : main

  • 版本: 1

  • 功能: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

该操作符的这个版本从版本 1 开始可用

概述

沿着多个轴生成输入张量的切片。类似于 numpy: https://numpy.com.cn/doc/stable/reference/routines.indexing.html 切片使用 axesstartsends 属性来指定轴列表中每个轴的开始和结束维度,它使用此信息来切片输入 data 张量。如果为任何开始或结束索引传递负值,则表示该维度结束前的元素数量。如果传递给开始或结束的值大于 n(该维度中的元素数量),则表示 n。对于切片到未知大小的维度的末尾,建议传入 INT_MAX。如果省略了 axes,则将其设置为 [0, ..., ndim-1]。示例 1:data = [ [1, 2, 3, 4], [5, 6, 7, 8], ] axes = [0, 1] starts = [1, 0] ends = [2, 3] result = [ [5, 6, 7], ] 示例 2:data = [ [1, 2, 3, 4], [5, 6, 7, 8], ] starts = [0, 1] ends = [-1, 1000] result = [ [2, 3, 4], ]

属性

  • axes - INTS :

    startsends 应用到的轴。它是可选的。如果不存在,将被视为 [0, 1, …, len(starts) - 1]。

  • ends - INTS (必填)

    axes` 中对应轴的结束索引(不包含)

  • starts - INTS (必填)

    axes 中对应轴的开始索引

输入

  • 数据 (异构) - T

    要从中提取切片的數據張量。

输出

  • 输出 (异构) - T

    切片后的数据张量。

类型约束

  • T in ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。