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 类型