CumSum

CumSum - 14

版本

  • 名称: CumSum (GitHub)

  • : main

  • 起始版本: 14

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

该版本的算子自 版本 14 起可用。

摘要

沿给定轴对输入元素执行累加和。默认情况下,它将执行包含性求和,这意味着第一个元素会原样复制。通过 exclusive 属性,此行为可以更改为排除第一个元素。它也可以沿轴的相反方向执行求和。为此,请将 reverse 属性设置为 1。

示例

input_x = [1, 2, 3]
axis=0
output = [1, 3, 6]
exclusive=1
output = [0, 1, 3]
exclusive=0
reverse=1
output = [6, 5, 3]
exclusive=1
reverse=1
output = [5, 3, 0]

属性

  • exclusive - INT (默认值为 '0')

    如果设置为 1,将返回不包含顶部元素的排他性求和。换句话说,如果设置为 1,则第 j 个输出元素将是前 (j-1) 个元素的总和。否则,它将是前 j 个元素的总和。

  • reverse - INT (默认值为 '0')

    如果设置为 1,将按相反方向执行求和。

输入

  • x (异构) - T

    要处理的输入张量。

  • axis (异构) - T2

    一个 0-D 张量。必须在 [-rank(x), rank(x)-1] 范围内。负值表示从后往前计算维度。

输出

  • y (异构) - T

    与 'x' 类型相同的输出张量,包含 x 元素的累积和

类型约束

  • T 必须是 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16), tensor(int32), tensor(int64), tensor(uint32), tensor(uint64) )

    将输入和输出类型限制为高精度数值张量。

  • T2 必须是 ( tensor(int32), tensor(int64) )

    轴张量只能是 int32 或 int64 类型

CumSum - 11

版本

  • 名称: CumSum (GitHub)

  • : main

  • 起始版本: 11

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

该版本的算子自 版本 11 起可用。

摘要

沿给定轴对输入元素执行累加和。默认情况下,它将执行包含性求和,这意味着第一个元素会原样复制。通过 exclusive 属性,此行为可以更改为排除第一个元素。它也可以沿轴的相反方向执行求和。为此,请将 reverse 属性设置为 1。

示例

input_x = [1, 2, 3]
axis=0
output = [1, 3, 6]
exclusive=1
output = [0, 1, 3]
exclusive=0
reverse=1
output = [6, 5, 3]
exclusive=1
reverse=1
output = [5, 3, 0]

属性

  • exclusive - INT (默认值为 '0')

    如果设置为 1,将返回不包含顶部元素的排他性求和。换句话说,如果设置为 1,则第 j 个输出元素将是前 (j-1) 个元素的总和。否则,它将是前 j 个元素的总和。

  • reverse - INT (默认值为 '0')

    如果设置为 1,将按相反方向执行求和。

输入

  • x (异构) - T

    要处理的输入张量。

  • axis (异构) - T2

    一个 0-D 张量。必须在 [-rank(x), rank(x)-1] 范围内。负值表示从后往前计算维度。

输出

  • y (异构) - T

    与 'x' 类型相同的输出张量,包含 x 元素的累积和

类型约束

  • T 必须是 ( tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(uint32), tensor(uint64) )

    输入可以是任何张量类型。

  • T2 必须是 ( tensor(int32), tensor(int64) )

    轴张量只能是 int32 或 int64 类型