Loading

2022年3月4日
PyTorch 中 torch.matmul() 函数的文档详解
摘要： 官方文档 torch.matmul() 函数几乎可以用于所有矩阵/向量相乘的情况，其乘法规则视参与乘法的两个张量的维度而定。 关于 PyTorch 中的其他乘法函数可以看这篇博文，有助于下面各种乘法的理解。 torch.matmul() 将两个张量相乘划分成了五种情形：一维 × 一维、二维 × 二维 阅读全文
posted @ 2022-03-04 10:11 Lowell_liu 阅读(9078) 评论(0) 推荐(5) 编辑