Gemm¶
Gemm - 13¶
版本¶
名称: Gemm (GitHub)
域:
main起始版本:
13函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的运算符自 版本 13 起可用。
摘要¶
通用矩阵乘法: https://en.wikipedia.org/wiki/Basic_Linear_Algebra_Subprograms#Level_3
A’ = 如果 transA 为真则转置(A) 否则 A
B’ = 如果 transB 为真则转置(B) 否则 B
计算 Y = alpha * A’ * B’ + beta * C，其中输入张量 A 的形状为 (M, K) 或 (K, M)，输入张量 B 的形状为 (K, N) 或 (N, K)，输入张量 C 可广播到形状 (M, N)，输出张量 Y 的形状为 (M, N)。如果属性 transA 非零，A 将在计算前转置，B 和 transB 亦同。此运算符支持单向广播（张量 C 应能单向广播到张量 A * B）；更多详情请查阅 ONNX 中的广播。此运算符具有可选输入/输出。有关可选参数表示的更多详情，请参见 ONNX IR。可以使用空字符串代替实际参数名称来指示缺失的参数。末尾的可选参数（后面没有出现参数的参数）也可以简单省略。
属性¶
alpha - FLOAT (默认值为
'1.0')输入张量 A * B 乘积的标量乘数。
beta - FLOAT (默认值为
'1.0')输入张量 C 的标量乘数。
transA - INT (默认值为
'0')A 是否应该转置
transB - INT (默认值为
'0')B 是否应该转置
输入¶
2 到 3 个输入之间。
A (异构) - T
输入张量 A。如果 transA 为 0，A 的形状应为 (M, K)；如果 transA 非零，则为 (K, M)。
B (异构) - T
输入张量 B。如果 transB 为 0，B 的形状应为 (K, N)；如果 transB 非零，则为 (N, K)。
C (可选，异构) - T
可选输入张量 C。如果未指定，则计算如同 C 是标量 0 一样进行。C 的形状应能单向广播到 (M, N)。
输出¶
Y (异构) - T
形状为 (M, N) 的输出张量。
类型约束¶
T 包含 (
tensor(bfloat16),tensor(double),tensor(float),tensor(float16),tensor(int32),tensor(int64),tensor(uint32),tensor(uint64))将输入和输出类型限制为浮点/整数张量。
Gemm - 11¶
版本¶
名称: Gemm (GitHub)
域:
main起始版本:
11函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的运算符自 版本 11 起可用。
摘要¶
通用矩阵乘法: https://en.wikipedia.org/wiki/Basic_Linear_Algebra_Subprograms#Level_3
A’ = 如果 transA 为真则转置(A) 否则 A
B’ = 如果 transB 为真则转置(B) 否则 B
计算 Y = alpha * A’ * B’ + beta * C，其中输入张量 A 的形状为 (M, K) 或 (K, M)，输入张量 B 的形状为 (K, N) 或 (N, K)，输入张量 C 可广播到形状 (M, N)，输出张量 Y 的形状为 (M, N)。如果属性 transA 非零，A 将在计算前转置，B 和 transB 亦同。此运算符支持单向广播（张量 C 应能单向广播到张量 A * B）；更多详情请查阅 ONNX 中的广播。此运算符具有可选输入/输出。有关可选参数表示的更多详情，请参见 ONNX IR。可以使用空字符串代替实际参数名称来指示缺失的参数。末尾的可选参数（后面没有出现参数的参数）也可以简单省略。
属性¶
alpha - FLOAT (默认值为
'1.0')输入张量 A * B 乘积的标量乘数。
beta - FLOAT (默认值为
'1.0')输入张量 C 的标量乘数。
transA - INT (默认值为
'0')A 是否应该转置
transB - INT (默认值为
'0')B 是否应该转置
输入¶
2 到 3 个输入之间。
A (异构) - T
输入张量 A。如果 transA 为 0，A 的形状应为 (M, K)；如果 transA 非零，则为 (K, M)。
B (异构) - T
输入张量 B。如果 transB 为 0，B 的形状应为 (K, N)；如果 transB 非零，则为 (N, K)。
C (可选，异构) - T
可选输入张量 C。如果未指定，则计算如同 C 是标量 0 一样进行。C 的形状应能单向广播到 (M, N)。
输出¶
Y (异构) - T
形状为 (M, N) 的输出张量。
类型约束¶
T 包含 (
tensor(double),tensor(float),tensor(float16),tensor(int32),tensor(int64),tensor(uint32),tensor(uint64))将输入和输出类型限制为浮点/整数张量。
Gemm - 9¶
版本¶
名称: Gemm (GitHub)
域:
main起始版本:
9函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的运算符自 版本 9 起可用。
摘要¶
通用矩阵乘法: https://en.wikipedia.org/wiki/Basic_Linear_Algebra_Subprograms#Level_3
A’ = 如果 transA 为真则转置(A) 否则 A
B’ = 如果 transB 为真则转置(B) 否则 B
计算 Y = alpha * A’ * B’ + beta * C，其中输入张量 A 的形状为 (M, K) 或 (K, M)，输入张量 B 的形状为 (K, N) 或 (N, K)，输入张量 C 可广播到形状 (M, N)，输出张量 Y 的形状为 (M, N)。如果属性 transA 非零，A 将在计算前转置，B 和 transB 亦同。此运算符支持单向广播（张量 C 应能单向广播到张量 A * B）；更多详情请查阅 ONNX 中的广播。
属性¶
alpha - FLOAT (默认值为
'1.0')输入张量 A * B 乘积的标量乘数。
beta - FLOAT (默认值为
'1.0')输入张量 C 的标量乘数。
transA - INT (默认值为
'0')A 是否应该转置
transB - INT (默认值为
'0')B 是否应该转置
输入¶
A (异构) - T
输入张量 A。如果 transA 为 0，A 的形状应为 (M, K)；如果 transA 非零，则为 (K, M)。
B (异构) - T
输入张量 B。如果 transB 为 0，B 的形状应为 (K, N)；如果 transB 非零，则为 (N, K)。
C (异构) - T
输入张量 C。C 的形状应能单向广播到 (M, N)。
输出¶
Y (异构) - T
形状为 (M, N) 的输出张量。
类型约束¶
T 包含 (
tensor(double),tensor(float),tensor(float16),tensor(int32),tensor(int64),tensor(uint32),tensor(uint64))将输入和输出类型限制为浮点/整数张量。
Gemm - 7¶
版本¶
名称: Gemm (GitHub)
域:
main起始版本:
7函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的操作符自版本 7 起可用。
摘要¶
通用矩阵乘法: https://en.wikipedia.org/wiki/Basic_Linear_Algebra_Subprograms#Level_3
A’ = 如果 transA 为真则转置(A) 否则 A
B’ = 如果 transB 为真则转置(B) 否则 B
计算 Y = alpha * A’ * B’ + beta * C，其中输入张量 A 的形状为 (M, K) 或 (K, M)，输入张量 B 的形状为 (K, N) 或 (N, K)，输入张量 C 可广播到形状 (M, N)，输出张量 Y 的形状为 (M, N)。如果属性 transA 非零，A 将在计算前转置，B 和 transB 亦同。此运算符支持单向广播（张量 C 应能单向广播到张量 A * B）；更多详情请查阅 ONNX 中的广播。
属性¶
alpha - FLOAT (默认值为
'1.0')输入张量 A * B 乘积的标量乘数。
beta - FLOAT (默认值为
'1.0')输入张量 C 的标量乘数。
transA - INT (默认值为
'0')A 是否应该转置
transB - INT (默认值为
'0')B 是否应该转置
输入¶
A (异构) - T
输入张量 A。如果 transA 为 0，A 的形状应为 (M, K)；如果 transA 非零，则为 (K, M)。
B (异构) - T
输入张量 B。如果 transB 为 0，B 的形状应为 (K, N)；如果 transB 非零，则为 (N, K)。
C (异构) - T
输入张量 C。C 的形状应能单向广播到 (M, N)。
输出¶
Y (异构) - T
形状为 (M, N) 的输出张量。
类型约束¶
T 在 (
tensor(double),tensor(float),tensor(float16))将输入和输出类型限制为浮点张量。
Gemm - 6¶
版本¶
名称: Gemm (GitHub)
域:
main起始版本:
6函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的操作符自版本 6 起可用。
摘要¶
通用矩阵乘法: https://en.wikipedia.org/wiki/Basic_Linear_Algebra_Subprograms#Level_3 计算 Y = alpha * A * B + beta * C，其中输入张量 A 的维度为 (M X K)，输入张量 B 的维度为 (K X N)，输入张量 C 和输出张量 Y 的维度为 (M X N)。如果属性 broadcast 非零，输入张量 C 将被广播以匹配维度要求。如果属性 transA 非零，A 将在计算前转置，B 和 transB 亦同。
属性¶
alpha - FLOAT (默认值为
'1.0')输入张量 A * B 乘积的标量乘数，默认值为 1.0。
beta - FLOAT (默认值为
'1.0')输入张量 C 的标量乘数，默认值为 1.0。
broadcast - INT（默认为
'0'）C 是否应该广播
transA - INT (默认值为
'0')A 是否应该转置
transB - INT (默认值为
'0')B 是否应该转置
输入¶
A (异构) - T
输入张量 A
B (异构) - T
输入张量 B
C (异构) - T
输入张量 C
输出¶
Y (异构) - T
输出张量。
类型约束¶
T 在 (
tensor(double),tensor(float),tensor(float16))将输入和输出类型限制为浮点张量。
Gemm - 1¶
版本¶
名称: Gemm (GitHub)
域:
main起始版本:
1函数:
False支持级别:
SupportType.COMMON形状推断:
False
此版本的运算符自 版本 1 起可用。
摘要¶
通用矩阵乘法: https://en.wikipedia.org/wiki/Basic_Linear_Algebra_Subprograms#Level_3 计算 Y = alpha * A * B + beta * C，其中输入张量 A 的维度为 (M X K)，输入张量 B 的维度为 (K X N)，输入张量 C 和输出张量 Y 的维度为 (M X N)。如果属性 broadcast 非零，输入张量 C 将被广播以匹配维度要求。如果属性 transA 非零，A 将在计算前转置，B 和 transB 亦同。
属性¶
alpha - FLOAT (默认值为
'1.0')输入张量 A * B 乘积的标量乘数，默认值为 1.0。
beta - FLOAT (默认值为
'1.0')输入张量 C 的标量乘数，默认值为 1.0。
broadcast - INT（默认为
'0'）C 是否应该广播
transA - INT (默认值为
'0')A 是否应该转置
transB - INT (默认值为
'0')B 是否应该转置
输入¶
A (异构) - T
输入张量 A
B (异构) - T
输入张量 B
C (异构) - T
输入张量 C，可以就地修改。
输出¶
Y (异构) - T
输出张量。
类型约束¶
T 在 (
tensor(double),tensor(float),tensor(float16))将输入和输出类型限制为浮点张量。