Pytorch torch.mul() 和 torch.mm() 的区别

Pytorch torch.mul() 和 torch.mm() 的区别

torch.mul()

矩阵点乘：要求进行运算的两矩阵维度相同。这里用3×3矩阵来举例说明。
$$\begin{gathered} A=\left[\begin{array}{ccc}{a}_{11}& a_{12}& a_{13}\\ a_{21}& a_{22}& a_{23}\\ a_{31}& a_{32}& a_{33}\end{array}\right] \\ B=\left[\begin{array}{ccc}{b}_{11}& b_{12}& b_{13}\\ b_{21}& b_{22}& b_{23}\\ b_{31}& b_{32}& b_{33}\end{array}\right] \end{gathered}$$
两者点乘的结果为
$$C=\left[\begin{array}{ccc}{a}_{11}\ast b_{11}& a_{12}\ast b_{12}& a_{13}\ast b_{13}\\ a_{21}\ast b_{21}& a_{22}\ast b_{22}& a_{23}\ast b_{23}\\ a_{31}\ast b_{31}& a_{32}\ast b_{32}& a_{33}\ast b_{33}\end{array}\right]$$
代码实现如下：

A = np.array([[1, 1, 1], [1, 1, 1], [1, 1, 1]])
B = np.array([[1, 1, 1], [1, 1, 1], [1, 1, 1]])
torch.mul(torch.tensor(A), torch.tensor(B))

结果如下：

torch.mm()

矩阵相乘：要求进行运算的两矩阵维度相同。这里用3×3矩阵来举例说明。
$$\begin{gathered} A=\left[\begin{array}{ccc}{a}_{11}& a_{12}& a_{13}\\ a_{21}& a_{22}& a_{23}\\ a_{31}& a_{32}& a_{33}\end{array}\right] \\ B=\left[\begin{array}{ccc}{b}_{11}& b_{12}& b_{13}\\ b_{21}& b_{22}& b_{23}\\ b_{31}& b_{32}& b_{33}\end{array}\right] \end{gathered}$$
两者相乘的结果为
$$C=\left[\begin{array}{ccc}{c}_{11}& c_{12}& c_{13}\\ c_{21}& c_{22}& c_{23}\\ c_{31}& c_{32}& c_{33}\end{array}\right]$$

$$c_{ij}=\sum _{k=1}^3{a}_{ik}{b}_{kj}$$

代码实现如下：

A = np.array([[1, 1, 1], [1, 1, 1], [1, 1, 1]])
B = np.array([[1, 1, 1], [1, 1, 1], [1, 1, 1]])
torch.mm(torch.tensor(A), torch.tensor(B))

结果如下：