高位矩阵的乘法为什么只作用于后两位,前面必须保持一致?
来源:3-7 Tensor的算术运算

weixin_慕UI0052680
2022-11-26
高维矩阵乘法为什么只作用于后两位,前面的位数到要保持一致?
写回答
1回答
-
会写代码的好厨师
2022-12-14
这个约束和matmul 这个算子的实现有关系。对于后面两纬要可以进行矩阵运算。对于前面的维度,需要对齐,否则会进行broadcast。
00
相似问题
高位矩阵的乘法为什么只作用于后两位,前面必须保持一致?
来源:3-7 Tensor的算术运算
weixin_慕UI0052680
2022-11-26
高维矩阵乘法为什么只作用于后两位,前面的位数到要保持一致?
1回答
会写代码的好厨师
2022-12-14
这个约束和matmul 这个算子的实现有关系。对于后面两纬要可以进行矩阵运算。对于前面的维度,需要对齐,否则会进行broadcast。
相似问题