ex_2_14.py

""" 为了能够现实下列代码的执行效果，请在安装PyTorch之后，在Python交互命令行界面，
    即在系统命令行下输入python这个命令回车后，在>>>提示符后执行下列代码
    （#号及其后面内容为注释，可以忽略）
"""

import torch

a = torch.randn(3,4) # 建立一个3×4的张量
b = torch.randn(4,3) # 建立一个4×3的张量
torch.mm(a,b) # 矩阵乘法，调用函数，返回3×3的矩阵乘积
a.mm(b) # 矩阵乘法，内置方法
a@b # 矩阵乘法，@运算符号
a = torch.randn(2,3,4) # 建立一个大小为2×3×4的张量
b = torch.randn(2,4,3) # 建立一个张量，大小为2×4×3
torch.bmm(a,b) # （迷你）批次矩阵乘法，返回结果为2×3×3，函数形式
a.bmm(b) # 同上乘法，内置方法形式
a@b # 运算符号形式，根据输入张量的形状决定调用批次矩阵乘法