torch.matmul

torch.matmul#

torch.matmul(input, other)

计算两个张量input和other的矩阵乘积,在PyTorch中可以通过torch.matmul函数实现

注意:matmul函数没有强制规定维度和大小,可以用利用广播机制进行不同维度的相乘操作。

import torch    
w = torch.tensor([2, -3.4])
X = torch.normal(0, 1, (1000, len(w)))
y = torch.matmul(X, w) + 4.2
y
tensor([ 1.5155e+00,  2.5159e+00, -4.7721e-01,  4.2631e+00,  5.5571e+00,
        -2.0366e+00,  8.8095e+00,  4.3449e+00, -6.3710e-03,  3.1130e+00,
         2.5257e+00,  6.8218e+00,  4.0516e+00,  1.3895e+00,  3.2285e+00,
         2.2649e+00, -5.6966e+00,  4.3563e+00,  4.1608e+00,  3.9191e+00,
         2.9248e+00,  1.4283e+00,  7.0282e+00,  7.8112e+00,  2.2465e+00,
         1.6021e+00,  7.5424e+00,  7.4619e+00,  1.1257e+00,  1.2088e+01,
        -7.1860e-01,  2.2883e+00,  2.8822e-01,  6.7184e+00,  2.8423e+00,
         1.2077e+01, -4.4317e+00, -2.0772e+00,  6.7896e+00,  6.5468e-01,
         4.8400e+00,  4.5392e-01,  5.2321e+00,  3.0468e+00,  1.2087e+01,
         1.2976e+00,  9.2901e+00,  7.6077e+00,  1.2924e+00,  6.7478e+00,
         3.6524e+00, -1.5192e+00,  2.6931e-01,  8.7465e+00,  4.0979e+00,
         3.1348e+00,  1.2177e+00, -2.2698e+00,  6.3479e-01,  5.6775e+00,
         5.4070e-02,  5.6914e+00,  2.0498e+00, -1.3071e+00,  4.0745e+00,
         5.9004e+00,  3.4423e+00,  4.2828e+00,  8.9808e+00,  3.2041e+00,
         9.9864e+00,  5.5425e+00,  2.6869e+00,  6.7037e+00,  1.9481e+00,
         2.6330e+00,  6.6180e+00,  9.6269e+00,  5.8402e+00,  3.4887e+00,
         6.7783e+00,  3.7153e-01,  5.6369e+00, -3.7723e+00,  4.5732e+00,
         5.4933e+00,  8.7657e+00, -1.4103e+00,  4.7612e-01,  1.1248e+00,
         1.2473e+01,  8.0082e+00,  3.7845e+00,  6.3955e+00,  1.7461e+00,
         1.3758e+00, -1.4982e+00, -8.5298e-01,  5.8901e+00,  6.1994e+00,
         8.0371e+00,  2.5930e+00,  2.1342e+00,  6.7650e+00,  7.7315e+00,
         4.8874e+00, -5.1029e+00, -2.9486e+00, -1.7192e+00, -8.7201e-01,
         3.3884e+00,  1.0255e+01,  5.1648e+00,  1.3463e+01, -6.1218e-01,
         7.2419e+00,  1.0668e+01,  3.6321e+00,  8.2857e+00,  3.7653e+00,
         9.8280e+00,  6.7659e+00, -4.2032e+00, -3.6893e-01,  8.6492e+00,
         2.9355e+00, -5.6927e-01,  2.8100e+00, -2.6885e+00,  8.5532e-01,
        -2.2902e+00,  4.2111e+00, -6.4140e-01,  2.8952e+00,  9.0203e+00,
         7.3646e+00,  3.8389e+00,  8.6855e+00,  8.2455e+00,  9.5468e+00,
         2.4407e+00,  8.1316e+00,  5.2892e+00,  4.0698e+00,  9.6687e+00,
        -7.0542e-02,  4.1158e+00,  5.5048e+00,  4.1971e+00,  4.7919e+00,
         9.2749e+00,  4.5310e+00,  9.1122e+00,  2.0163e+00,  8.3804e+00,
         5.7845e+00,  8.3384e+00,  1.1779e+01,  9.4574e+00,  4.2705e+00,
         3.0511e+00,  9.0905e+00,  9.7384e+00, -3.7845e+00,  1.0141e+01,
         1.0589e+01,  7.5515e+00,  2.4432e+00,  7.4614e+00,  4.6743e+00,
         3.0650e+00,  3.6797e+00,  2.5105e+00, -2.1548e+00, -4.2095e-01,
         7.3590e+00,  1.7389e+00, -5.5587e+00,  5.1342e+00,  4.1943e+00,
         3.0771e+00, -1.6831e+00,  3.4473e+00,  9.9928e+00,  6.2434e+00,
         2.6036e-01,  9.0087e+00,  7.6408e+00,  2.5850e+00, -4.8009e+00,
         2.0166e+00, -2.8029e+00,  4.5983e+00,  8.3093e+00,  4.5704e-01,
         4.5585e+00,  8.5441e+00,  6.2608e+00,  8.1324e+00,  4.6281e+00,
         5.9037e+00,  1.2701e+01,  5.8832e-01,  2.4913e+00,  1.1748e+01,
         1.0847e-01,  5.3911e+00,  1.0864e-01,  2.3300e+00,  8.7759e+00,
         1.7019e+00,  7.0659e-01,  1.0519e+01, -1.0123e+00, -7.3321e-02,
        -1.2179e+00,  1.1622e+01,  8.5840e+00,  1.0250e+01,  3.4501e+00,
         4.6669e-01,  1.1260e+01,  4.1858e+00,  1.8801e+00, -2.5138e+00,
         4.6483e+00,  1.8272e+00,  2.9033e+00,  3.6973e+00, -1.7329e+00,
         2.3129e+00,  1.5406e+00,  1.0068e+01,  1.6656e+00, -3.7109e+00,
         4.5971e+00,  3.2986e+00,  6.7161e+00,  3.1461e+00, -1.3196e+00,
         8.2856e-01,  6.8960e+00,  7.0238e+00,  2.4367e+00,  4.9079e+00,
         5.9120e+00, -1.2816e+00,  2.9537e+00,  4.1722e+00,  1.1391e+01,
         4.3921e+00,  1.0580e+01,  4.2049e+00,  4.3067e+00,  5.2432e+00,
         2.0285e+00,  4.9968e+00,  1.4071e+00, -4.0973e-01,  4.4319e+00,
        -1.3866e+00,  1.7285e+00,  2.4017e+00,  2.5505e+00, -1.7765e+00,
         6.4156e+00, -3.8597e+00,  3.0522e-01,  8.5046e+00, -3.0596e+00,
         8.5585e-01,  7.1339e+00,  6.8515e+00,  7.4323e+00,  4.9080e+00,
         3.2506e+00,  9.6203e+00, -1.9370e+00, -2.5691e+00,  4.5410e+00,
         1.6057e+00,  8.2213e+00,  5.9239e+00,  8.2826e-01,  1.0678e+01,
        -1.9585e+00,  6.8751e+00,  9.7065e+00, -1.0770e+00,  2.0853e+00,
         5.1298e+00,  5.7724e+00,  3.8012e+00,  8.1134e-01,  9.8118e+00,
         2.5886e+00, -2.6686e+00,  3.7005e+00,  4.9225e+00,  3.3569e+00,
         1.0588e+00,  8.8949e+00,  4.7218e+00,  4.4370e+00,  2.2404e+00,
         3.0963e+00,  1.2403e+01,  7.7327e+00,  2.5726e+00,  7.4017e+00,
         2.9951e+00,  4.8737e+00,  4.4579e+00,  2.2162e+00,  7.4309e+00,
         5.2681e+00, -4.2074e+00,  8.8411e+00,  6.5685e+00,  1.0565e+01,
         6.4361e+00,  5.6568e+00, -6.6984e-01,  9.7495e+00,  4.6765e+00,
         4.7174e+00,  6.0274e+00,  7.3052e-01,  2.8634e+00,  4.3409e+00,
         5.0563e+00,  6.6374e-01,  4.6477e+00,  8.9529e+00,  7.2615e+00,
         1.7798e+00,  7.4227e+00,  8.2483e+00,  9.6006e+00,  3.7333e+00,
         1.9035e+00,  8.9252e+00, -2.9921e+00, -3.1988e+00,  1.5569e+00,
         4.5282e+00,  1.3786e+00,  4.8587e+00,  4.6989e+00,  8.1235e+00,
         1.7509e+00,  3.1115e-01,  7.6582e+00,  5.7561e+00, -6.8837e-01,
        -2.2680e+00,  4.6194e+00,  4.3493e+00,  2.2307e+00,  6.4572e+00,
         3.3157e+00,  1.1704e+01,  1.0427e+01,  1.6671e+00, -8.5014e-01,
         7.8442e+00, -1.1318e+00,  4.5835e+00,  3.0543e+00,  9.2412e+00,
        -8.1021e-01,  4.8855e+00,  7.7519e-02,  9.6085e+00,  1.1604e+01,
         8.0013e+00,  3.7215e+00,  1.6337e+00,  9.1372e+00,  1.0556e+01,
         7.0946e+00,  9.4726e+00,  3.1591e+00,  1.6834e+00,  7.4092e+00,
         7.6926e+00,  2.8496e+00,  7.6874e+00,  7.1784e+00,  4.4475e+00,
        -2.4069e+00,  1.8774e+00,  2.7238e+00,  6.2191e+00, -3.5535e+00,
         1.5226e+01, -4.1117e+00,  3.0818e+00, -1.7982e+00,  2.0360e+00,
         5.7937e+00,  1.1358e+01,  6.6973e+00,  1.0070e+00,  6.6890e+00,
         2.2247e+00,  4.7083e+00,  7.0884e-01,  4.9310e-01,  6.4377e+00,
         2.3786e+00,  1.1806e-01, -3.3875e+00,  4.8865e+00,  4.3418e+00,
         3.5571e+00,  9.6810e+00,  6.7636e+00,  1.2701e+01,  5.6403e+00,
         5.5436e+00,  5.1805e-01,  7.7228e+00,  4.7175e+00,  1.7305e+00,
         6.8131e-01,  1.9667e+00, -2.5653e+00, -3.2664e+00,  1.1711e+01,
         9.7890e+00,  2.5587e+00,  6.3787e+00,  8.3034e+00,  5.5790e+00,
         3.8287e-01,  7.8141e+00,  9.8087e+00,  8.9269e-01,  5.6734e+00,
         7.6379e+00,  1.3314e+01,  4.0592e+00,  5.2090e+00,  4.1336e+00,
         5.9095e+00, -1.6845e+00,  5.3227e+00,  4.0723e+00,  8.4984e+00,
         5.6931e+00,  5.2018e+00,  1.0037e+01,  6.8649e+00,  6.3762e+00,
         5.5660e+00,  9.4641e+00,  2.7494e+00,  2.9750e+00, -9.4422e-01,
         2.7011e+00, -4.2667e-01, -1.0258e+00,  7.5701e+00,  6.1535e+00,
         2.8489e+00,  5.1939e+00,  2.8975e+00,  6.6322e+00,  5.5360e-01,
         8.9918e+00,  1.5407e+01,  1.5847e+00,  8.6042e+00,  4.3903e+00,
         1.0463e+01,  8.3233e+00,  9.1589e+00,  7.0592e+00,  1.3794e+00,
         2.4175e+00,  7.4523e+00,  1.8596e+00,  9.7404e+00,  5.8474e+00,
         4.6569e+00,  5.1460e+00,  6.2494e-01,  5.0916e+00,  7.7493e+00,
         2.2050e+00, -4.7043e+00, -2.5267e+00,  9.0437e+00,  3.2816e+00,
         1.8058e+00, -1.2188e+00,  9.5780e+00,  2.1029e-02,  2.4650e+00,
         1.0103e+01,  5.3098e+00,  7.7253e+00,  6.5316e+00,  1.1138e+01,
         1.7885e+00,  1.3061e+01,  6.2880e+00, -2.8147e+00, -5.1226e-01,
         8.2079e+00, -2.9254e+00, -4.4346e+00,  8.6043e+00,  4.6594e+00,
         4.0115e+00,  8.4843e+00,  2.5802e+00,  2.8044e+00,  1.2259e+00,
         1.0287e+01,  1.2378e+01,  5.5268e+00,  6.7247e-01,  4.7194e+00,
        -3.5654e+00, -4.2324e+00, -1.9214e-02,  3.9706e+00, -5.7546e-01,
         1.1322e+00,  3.3077e+00, -5.7408e+00,  2.7227e+00,  3.3966e+00,
         4.8431e+00,  7.5766e+00,  3.6355e+00,  4.1795e+00,  8.5443e+00,
         2.6548e+00,  1.1224e+01, -2.6587e+00,  7.8720e+00,  2.9656e+00,
         4.2518e-01,  9.0950e+00,  6.5125e+00, -2.5454e+00,  1.3034e+01,
         7.2486e+00,  1.1285e+00,  5.9891e-02, -4.7399e-01,  5.5526e+00,
         1.5804e+01,  2.7533e+00,  1.4741e-01,  8.7893e-01,  5.4333e-01,
         8.6456e+00,  5.3563e+00,  4.5438e+00,  1.1746e+00, -2.3008e+00,
         6.2561e+00,  2.5309e+00,  9.0806e+00,  8.7450e+00,  3.4887e+00,
         2.3344e+00,  6.5545e+00, -4.3733e-01,  2.3263e+00,  4.5804e+00,
        -1.7500e+00, -8.0607e-02,  3.0787e+00,  2.6709e+00,  6.6869e+00,
         3.2466e+00,  1.1100e+01,  1.8125e+00,  5.9496e+00,  2.3706e-01,
         2.1297e+00, -7.3636e-01,  9.2665e+00, -5.3181e-01,  8.3611e+00,
         1.3276e+00,  4.6655e+00,  7.8029e+00,  8.6678e+00,  4.0951e+00,
         1.9437e+00,  5.3442e+00,  9.2943e+00,  4.4699e+00,  5.6708e+00,
         1.0411e+01,  1.3265e+01,  3.0062e+00,  5.5413e+00, -3.1547e-01,
         1.4081e+01,  2.3923e-01,  7.6408e+00, -1.9675e+00,  6.6704e+00,
         2.3434e+00,  2.8340e+00,  2.5743e+00,  1.0425e+01,  7.7651e+00,
         4.2655e+00,  4.3229e+00,  3.3200e+00,  7.6975e+00,  5.2823e+00,
         2.2825e+00,  4.0018e+00,  1.2286e+00,  6.8803e+00,  8.0758e+00,
         4.2641e+00,  1.2905e+01,  3.6259e+00,  6.8471e+00,  6.7916e+00,
         3.5901e+00,  2.6847e-01,  6.3000e+00,  9.7851e+00, -2.8520e+00,
         1.2703e+00,  1.2381e+01,  3.6104e+00,  3.4345e+00,  6.4867e+00,
         6.9241e+00,  6.2413e+00,  3.1281e+00,  8.6509e+00,  7.9985e+00,
         3.5395e+00,  2.2762e+00,  1.0478e+01, -1.8401e+00,  3.3917e-01,
         2.4062e+00,  5.6498e-01, -3.4856e+00,  5.0841e+00,  2.4541e+00,
         2.4888e+00,  3.9738e+00,  3.1630e+00,  5.4614e+00,  3.0399e+00,
         9.0030e+00,  6.1623e+00,  1.0965e+01,  1.1671e+01,  5.7218e+00,
        -5.7584e-01,  2.7013e+00,  7.8328e+00,  1.0445e+00,  5.5469e+00,
         3.4884e+00, -1.7965e+00,  1.1557e+01,  4.1444e+00, -1.9666e+00,
         2.2697e+00, -1.6774e-02, -1.8478e+00,  3.3886e+00,  2.9302e+00,
         7.1722e+00,  6.2253e+00,  7.1344e+00, -1.3232e+00,  4.8268e+00,
         3.5138e+00,  3.5291e+00,  4.2307e+00,  9.5542e+00,  9.1950e-01,
         9.4876e+00,  6.6506e+00,  3.8162e+00,  4.6597e+00,  1.6232e+00,
         5.9657e+00,  8.5667e+00,  8.9427e-01,  5.4127e+00,  7.0024e+00,
         1.3799e-01,  8.9763e+00,  1.7245e+00,  5.1453e+00,  5.6977e+00,
         5.5139e-01,  3.3094e+00,  3.9420e-01,  6.8544e+00,  7.7775e+00,
         6.2660e+00,  2.7832e+00,  3.7762e+00, -5.2730e-01, -6.1577e-01,
         8.7663e+00,  7.4977e+00, -2.8144e+00,  8.5487e+00,  1.8861e+00,
         9.7686e+00,  2.3420e+00,  7.1857e+00, -1.2485e+00,  6.7817e+00,
         1.9415e+00,  1.2005e+01, -2.3292e+00,  9.3291e-01, -2.5506e+00,
         5.3573e+00,  4.4193e+00,  9.5143e+00,  8.1314e+00,  1.4508e+01,
         8.5992e-01,  8.7445e+00,  5.4032e+00,  6.1884e+00,  4.1102e+00,
         1.8841e+00,  2.0376e+00,  9.4014e-02,  3.2794e+00,  5.0239e+00,
         9.6984e+00, -1.4802e+00,  1.9785e+00,  2.4519e+00,  5.3263e+00,
         1.5201e+00, -3.3383e-01, -3.5468e+00,  6.5452e+00,  7.7473e+00,
         4.2667e+00,  5.0670e+00,  9.3995e+00,  2.8062e+00,  4.7074e+00,
         8.0939e+00,  5.4510e+00, -2.2368e+00,  3.2565e+00,  4.5355e+00,
         1.1706e+01,  2.8092e+00,  7.5148e-01,  8.8784e-01, -6.2732e+00,
         3.5047e+00,  9.7349e+00,  4.5564e+00,  3.1334e+00, -3.8440e+00,
         1.2370e+01,  1.1777e+00,  4.2689e+00,  3.5788e+00, -4.0398e-01,
         2.9643e+00, -1.0079e+00,  1.4452e+00,  2.7214e+00,  4.1228e+00,
         6.1217e+00,  2.7918e+00,  4.0705e-01,  1.4267e+01,  3.6822e+00,
        -4.6873e+00,  4.8190e+00,  2.5299e+00,  5.0762e+00, -6.3267e+00,
         1.7328e+00,  7.1314e+00,  6.0880e+00,  8.1776e-01,  3.7505e+00,
        -3.1931e+00,  5.7360e-01,  1.3243e+00,  3.7348e+00, -6.4673e+00,
         3.2012e+00,  5.0231e+00,  8.7408e+00,  9.9152e+00,  4.6285e+00,
         3.9734e+00,  4.3938e+00,  1.1135e+00,  3.0901e+00,  9.0204e+00,
         3.8358e+00,  6.9584e+00,  2.2618e+00,  3.0287e+00,  6.2825e+00,
         7.7375e+00,  2.2889e+00,  9.4978e+00,  1.5331e+00, -7.2899e-01,
         1.2033e+00,  2.7158e+00,  1.8642e+00,  1.8063e+00, -8.0998e-01,
        -2.4989e+00,  8.2907e+00,  4.8188e+00,  6.8195e+00,  4.0714e+00,
         9.2875e-01,  4.0919e+00,  4.6670e+00,  1.5665e+00,  7.7566e+00,
         7.1786e+00,  1.1154e+01,  8.2255e+00,  3.2416e+00,  5.0054e+00,
         3.5681e+00,  5.3130e+00,  6.5287e+00, -2.5260e+00,  4.2878e+00,
         6.4519e+00,  6.8854e+00,  2.3860e+00,  8.3893e+00,  6.0768e+00,
         6.4309e+00,  7.3918e+00,  8.5761e+00,  1.1136e+00,  4.1600e+00,
         6.6610e+00, -1.5055e+00, -5.6011e+00,  5.7012e+00,  4.1165e+00,
         6.5797e+00, -1.4508e+00, -1.1256e+00, -3.4919e+00,  6.2004e+00,
         1.0700e+01,  3.9573e+00,  4.5119e+00,  5.3501e+00,  8.0782e+00,
         5.6821e+00,  1.1259e+00,  2.4759e+00,  7.1893e+00, -6.1090e+00,
        -1.3448e+00,  1.0211e+01,  4.3035e+00,  5.4645e+00, -2.9056e+00,
         4.9030e+00,  6.4353e+00,  1.9762e+00,  8.3358e+00,  7.2138e+00,
        -7.7169e-01, -3.6802e+00,  2.0003e+00,  8.8170e+00,  1.3524e+00,
         1.7122e+00,  6.1526e-01,  8.4881e+00, -7.1586e-01,  2.4554e+00,
         7.0035e+00,  4.2372e-01,  7.5047e+00,  8.9237e-01,  9.5494e+00,
         2.6058e+00, -1.2327e-01,  2.4325e+00,  7.6158e+00,  6.7378e+00,
         3.2242e+00,  7.6589e+00,  7.9768e+00,  9.3375e+00,  3.1339e+00,
         5.6536e+00,  4.2912e+00, -2.3762e+00,  1.2843e+00, -3.0787e-01,
         7.4058e-01,  4.6719e-01,  7.6727e+00,  1.0692e+01,  1.1390e+01,
         9.5596e+00,  4.1170e-01, -1.0832e-01,  1.7154e+00,  2.0617e+00,
         5.4850e+00,  1.3598e+01,  1.8222e+00,  3.9711e+00,  4.0794e+00,
        -3.2162e+00, -8.1225e-01,  3.6451e+00,  6.3822e+00,  5.7103e+00,
        -1.4212e+00,  2.0552e+00, -1.4486e+00,  9.4252e+00,  6.3849e+00,
         4.7307e+00,  3.5159e+00,  7.0586e+00,  5.1030e-01,  4.6364e+00,
         3.5575e+00,  1.1610e+01,  4.8578e+00,  8.2146e-01,  1.2263e+00,
         9.6002e+00,  2.9163e+00,  3.9493e+00,  1.0038e+01,  3.1746e+00,
        -2.0964e+00, -1.8012e+00,  5.9107e+00,  2.0021e+00,  1.1764e+01,
         6.4404e+00,  5.7346e+00, -4.5888e+00,  4.2857e+00,  5.5137e+00,
         1.0847e+01,  5.4401e+00,  2.4594e+00,  1.0269e+00,  5.2835e+00,
         5.0214e+00,  6.2359e+00,  3.9857e+00,  7.6423e+00,  6.2848e+00,
         5.7548e+00,  5.5761e+00,  5.1107e+00,  7.3921e+00,  1.0115e+01,
         4.8001e+00,  4.2000e+00,  4.4438e+00,  3.0132e+00, -3.8930e+00,
         5.8161e+00,  3.7027e+00,  2.9252e+00,  7.7859e+00,  1.3582e+00,
         9.8997e+00,  4.5158e+00, -1.7127e+00,  5.6564e+00,  9.3475e+00])