高位矩阵的乘法为什么只作用于后两位,前面必须保持一致?

来源:3-7 Tensor的算术运算

weixin_慕UI0052680

2022-11-26

高维矩阵乘法为什么只作用于后两位,前面的位数到要保持一致?

写回答

1回答

会写代码的好厨师

2022-12-14

这个约束和matmul 这个算子的实现有关系。对于后面两纬要可以进行矩阵运算。对于前面的维度,需要对齐,否则会进行broadcast。

0
0

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 298 问题

查看课程