移除维度¶
移除维度 - 23¶
版本¶
名称: Squeeze (GitHub)
域:
main
始于版本:
23
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符 始于版本 23。
摘要¶
移除张量形状中的单维度条目。接受一个输入 axes
,其中包含要挤压的轴列表。如果未提供 axes
,将从形状中移除所有单维度。如果选择的轴的形状条目不等于一,则会引发错误。
输入¶
介于 1 到 2 个输入。
data (异构) - T
具有至少 max(dims) 维度的张量。
axes (可选, 异构) - tensor(int64)
表示要挤压的维度的整数的一维张量。负值表示从后向前计数维度。接受的范围是 [-r, r-1],其中 r = rank(data)。
输出¶
squeezed (异构) - T
数据与输入相同的重塑张量。
类型约束¶
T 属于 (
tensor(bfloat16)
,tensor(bool)
,tensor(complex128)
,tensor(complex64)
,tensor(double)
,tensor(float)
,tensor(float16)
,tensor(float4e2m1)
,tensor(float8e4m3fn)
,tensor(float8e4m3fnuz)
,tensor(float8e5m2)
,tensor(float8e5m2fnuz)
,tensor(int16)
,tensor(int32)
,tensor(int4)
,tensor(int64)
,tensor(int8)
,tensor(string)
,tensor(uint16)
,tensor(uint32)
,tensor(uint4)
,tensor(uint64)
,tensor(uint8)
)将输入和输出类型约束为 IRv11 及以前的所有张量类型。
移除维度 - 21¶
版本¶
名称: Squeeze (GitHub)
域:
main
始于版本:
21
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符 始于版本 21。
摘要¶
移除张量形状中的单维度条目。接受一个输入 axes
,其中包含要挤压的轴列表。如果未提供 axes
,将从形状中移除所有单维度。如果选择的轴的形状条目不等于一,则会引发错误。
输入¶
介于 1 到 2 个输入。
data (异构) - T
具有至少 max(dims) 维度的张量。
axes (可选, 异构) - tensor(int64)
表示要挤压的维度的整数列表。负值表示从后向前计数维度。接受的范围是 [-r, r-1],其中 r = rank(data)。
输出¶
squeezed (异构) - T
数据与输入相同的重塑张量。
类型约束¶
T 属于 (
tensor(bfloat16)
,tensor(bool)
,tensor(complex128)
,tensor(complex64)
,tensor(double)
,tensor(float)
,tensor(float16)
,tensor(float8e4m3fn)
,tensor(float8e4m3fnuz)
,tensor(float8e5m2)
,tensor(float8e5m2fnuz)
,tensor(int16)
,tensor(int32)
,tensor(int4)
,tensor(int64)
,tensor(int8)
,tensor(string)
,tensor(uint16)
,tensor(uint32)
,tensor(uint4)
,tensor(uint64)
,tensor(uint8)
)将输入和输出类型约束为 IRv10 及以前的所有张量类型。
移除维度 - 13¶
版本¶
名称: Squeeze (GitHub)
域:
main
始于版本:
13
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符 始于版本 13。
摘要¶
移除张量形状中的单维度条目。接受一个输入 axes
,其中包含要挤压的轴列表。如果未提供 axes
,将从形状中移除所有单维度。如果选择的轴的形状条目不等于一,则会引发错误。
输入¶
介于 1 到 2 个输入。
data (异构) - T
具有至少 max(dims) 维度的张量。
axes (可选, 异构) - tensor(int64)
表示要挤压的维度的整数列表。负值表示从后向前计数维度。接受的范围是 [-r, r-1],其中 r = rank(data)。
输出¶
squeezed (异构) - T
数据与输入相同的重塑张量。
类型约束¶
T 属于 (
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)
)将输入和输出类型约束为所有张量类型。
移除维度 - 11¶
版本¶
名称: Squeeze (GitHub)
域:
main
始于版本:
11
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符 始于版本 11。
摘要¶
移除张量形状中的单维度条目。接受一个参数 axes
,其中包含要挤压的轴列表。如果未提供 axes
,将从形状中移除所有单维度。如果选择的轴的形状条目不等于一,则会引发错误。
属性¶
axes - INTS :
表示要挤压的维度的整数列表。负值表示从后向前计数维度。接受的范围是 [-r, r-1],其中 r = rank(data)。
输入¶
data (异构) - T
具有至少 max(dims) 维度的张量。
输出¶
squeezed (异构) - T
数据与输入相同的重塑张量。
类型约束¶
T 属于 (
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)
)将输入和输出类型约束为所有张量类型。
移除维度 - 1¶
版本¶
名称: Squeeze (GitHub)
域:
main
始于版本:
1
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符 始于版本 1。
摘要¶
移除张量形状中的单维度条目。接受一个参数 axes
,其中包含要挤压的轴列表。如果未提供 axes
,将从形状中移除所有单维度。如果选择的轴的形状条目不等于一,则会引发错误。
属性¶
axes - INTS :
非负整数列表,指示要挤压的维度。
输入¶
data (异构) - T
具有至少 max(dims) 维度的张量。
输出¶
squeezed (异构) - T
数据与输入相同的重塑张量。
类型约束¶
T 属于 (
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)
)将输入和输出类型约束为所有张量类型。