其他分享
首页 > 其他分享> > torch.matmul

torch.matmul

作者:互联网

 

0阶张量是一个标量。

1阶张量是一个向量。

2阶张量是一个矩阵。

如果俩个张量都是一维度,输出的结果是标量相乘。

如果俩个张量都是矩阵,输出的结果是矩阵乘积。

如果第一个是二维,第二个是一维张量,输出结果将第一个参数每一行看成一个向量与第二个进行向量积。(只有他需要注意)

如果第一个是一维张量,第二个是二维张量,输出结果也是矩阵乘积。

标签:输出,torch,矩阵,张量,第二个,matmul,俩个,向量
来源: https://www.cnblogs.com/hahaah/p/14934011.html