Where¶
Where - 16¶
版本¶
名称: Where (GitHub)
域:
main自版本:
16函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的操作符自 版本 16 开始可用。
摘要¶
根据条件返回元素，来自 X 或 Y。Where 的行为类似于 numpy.where，带三个参数。
此操作符支持 多方向（即 NumPy 风格）广播；有关更多详细信息，请查看 ONNX 中的广播。
输入¶
condition (异构) - B
当为 True（非零）时，生成 X，否则生成 Y
X (异构) - T
在 condition 为 True 的索引处选定的值
Y (异构) - T
在 condition 为 False 的索引处选定的值
输出¶
output (异构) - T
形状等于 condition、X 和 Y 的广播形状的张量。
类型约束¶
B in (
tensor(bool))约束为布尔型张量。
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))将输入和输出类型约束为所有张量类型（包括 bfloat）。
Where - 9¶
版本¶
名称: Where (GitHub)
域:
main自版本:
9函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的操作符自 版本 9 开始可用。
摘要¶
根据条件返回元素，来自 X 或 Y。Where 的行为类似于 numpy.where，带三个参数。
此操作符支持 多方向（即 NumPy 风格）广播；有关更多详细信息，请查看 ONNX 中的广播。
输入¶
condition (异构) - B
当为 True（非零）时，生成 X，否则生成 Y
X (异构) - T
在 condition 为 True 的索引处选定的值
Y (异构) - T
在 condition 为 False 的索引处选定的值
输出¶
output (异构) - T
形状等于 condition、X 和 Y 的广播形状的张量。
类型约束¶
B in (
tensor(bool))约束为布尔型张量。
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))将输入和输出类型约束为所有张量类型。