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([ 9.3910e+00,  3.6463e+00,  4.5379e+00,  1.0628e+01, -3.0676e+00,
        -2.5783e+00,  3.8141e+00,  1.7285e-01,  1.2522e+01,  7.2375e+00,
         1.7479e+00,  6.1676e+00, -8.3922e+00, -6.8108e-01,  5.0356e+00,
         6.0544e+00,  1.9171e+00,  2.7552e+00,  3.5581e+00,  9.0512e+00,
         3.0291e-01,  2.3103e+00,  9.4529e+00,  3.4220e+00, -1.3745e-01,
         3.2791e+00,  4.5786e+00, -6.4339e-01,  1.4956e+00,  7.4200e+00,
         4.0195e-01, -2.4146e+00,  1.0840e+01, -4.9548e+00,  3.8174e+00,
         7.5059e+00,  3.9997e+00,  9.9165e+00,  5.4644e+00,  8.3569e+00,
         3.1072e+00,  8.0813e+00,  1.6971e+00, -1.6913e+00,  7.4097e-01,
         4.1224e-01,  4.6373e+00,  2.6921e+00,  2.4285e+00,  8.5896e+00,
         1.5555e+00,  7.6308e+00, -2.7533e-01,  6.8202e+00,  6.8630e-01,
         4.7311e+00,  9.7597e-01,  9.5642e-01,  1.7373e+00, -2.2339e-01,
         4.6181e+00,  5.1665e+00,  5.2801e+00,  1.7798e+00,  7.1100e+00,
         7.3755e+00,  9.6836e+00,  8.1975e+00,  3.5414e+00,  5.7395e+00,
         5.9851e+00,  6.6112e+00,  5.7456e+00,  4.4104e+00,  2.7605e+00,
         8.5292e-01,  4.0559e+00,  4.6134e+00, -9.0192e-01,  5.9827e+00,
         1.8702e+00,  8.7470e+00,  4.2595e-01,  6.6748e+00, -2.1460e-01,
        -1.8629e-01,  3.6129e+00,  6.0520e+00,  1.5594e+00,  1.2329e+00,
         1.1970e+01,  4.5980e+00,  4.9577e+00,  4.0492e-01,  5.9149e+00,
         6.5174e+00,  4.1315e+00,  5.9009e+00,  2.6229e+00,  1.1496e+01,
         4.1609e+00,  3.9184e+00,  9.5421e+00,  8.9377e+00,  9.7159e-01,
        -1.9294e+00, -3.9111e+00,  8.5929e+00,  3.4762e+00,  6.8613e-01,
         3.3297e+00,  3.3269e+00,  8.5882e+00,  5.7329e+00,  3.9571e+00,
         6.5203e+00,  2.4398e+00,  2.6186e+00,  4.1707e+00,  5.2026e+00,
         9.5651e+00,  4.2399e+00,  4.9583e+00,  8.3934e+00,  1.0565e+01,
         1.6533e-01,  2.9275e+00,  6.9534e+00,  1.0151e+01, -1.2081e+00,
        -1.7801e+00,  3.2868e+00, -1.1478e+00,  2.0885e+00,  2.4204e+00,
         6.3048e+00, -3.0923e+00,  4.0784e+00,  1.3370e+00,  1.7288e+00,
        -1.6170e-01, -1.0683e+00,  6.1754e+00,  4.2864e+00, -1.6352e+00,
         7.8316e+00,  1.5331e-01,  8.9665e+00,  9.9375e+00,  1.2125e+01,
         9.6601e+00,  5.1799e+00,  5.4900e+00,  8.8425e+00,  7.2321e+00,
         9.6677e+00,  6.0457e+00,  6.6652e+00,  2.0966e+00,  6.7963e+00,
         4.9738e+00,  3.1490e+00,  5.4160e+00,  6.2494e+00,  2.1244e+00,
        -2.3148e-01,  7.0334e+00,  1.5741e-01,  1.1140e+00,  8.9036e+00,
         1.0003e+00,  5.2605e+00,  2.5535e+00,  1.0168e+01,  2.8069e+00,
         7.7594e+00,  5.5368e+00,  5.3106e+00,  2.5612e+00,  3.2979e+00,
        -9.8089e-01,  8.5242e+00,  1.0118e+01, -9.6977e-02,  1.4716e+00,
         7.4304e+00,  1.0122e+01,  9.5628e-01,  3.6748e+00,  7.6551e+00,
         8.1954e+00,  4.4924e+00,  7.6705e+00,  9.1144e+00,  1.0225e+01,
         8.3587e+00,  6.0816e+00,  4.2159e+00,  3.8942e+00,  5.7537e+00,
         8.3982e+00,  9.9181e+00,  7.4471e+00,  3.9847e+00,  1.0824e+00,
         5.5926e+00,  6.4736e+00,  1.1681e+01,  2.4899e+00, -2.4025e+00,
         1.7043e-01, -2.6976e+00,  3.4318e+00, -4.9398e+00, -2.9566e-01,
         1.3957e+01,  1.2091e+00, -2.0703e+00,  1.1787e+01,  6.4638e+00,
         4.0884e+00,  6.8613e+00,  4.0049e+00,  5.2010e+00,  3.1350e+00,
         4.9716e-01,  7.7790e+00,  9.8551e-01,  7.8605e+00,  4.0283e+00,
         4.0308e+00,  7.2281e+00, -5.0458e-01, -2.0622e+00,  7.3526e+00,
        -9.4448e-01,  6.0255e+00,  2.9402e+00,  6.7431e+00,  1.2420e+00,
        -2.1775e+00, -1.4095e+00,  8.4745e+00,  9.2593e+00,  6.2099e+00,
         5.5066e+00,  9.5558e-01,  9.7803e+00,  8.3936e+00,  7.0249e+00,
         4.4783e+00,  5.6468e+00,  2.0118e+00, -2.5047e-01,  6.1456e+00,
         5.2985e+00,  1.6110e+01,  8.5549e+00,  9.9540e-01,  4.2503e+00,
         8.3387e+00, -1.7815e+00,  5.3926e+00,  5.4270e+00,  9.7477e+00,
         7.2832e-01,  4.2945e+00,  1.7867e+00,  9.0310e+00,  6.6146e+00,
        -1.6422e+00,  2.4932e+00,  1.7298e+00, -6.7810e+00,  7.5067e+00,
         1.5279e+00,  5.7550e+00,  1.4507e+00, -2.9787e-01,  2.6831e+00,
         4.3242e+00,  9.4379e+00, -5.1316e+00,  6.9031e+00,  6.9396e+00,
         1.0132e+01,  1.2478e+00, -4.4305e-01,  6.5438e-01,  2.2017e+00,
         1.2018e+01,  6.9626e-01,  6.7619e+00,  1.0420e+00,  1.3123e+00,
         4.4660e+00,  2.8742e+00,  5.9198e+00,  3.6371e+00,  4.9482e+00,
        -5.4419e+00,  2.6958e-01,  3.6337e+00,  1.5082e+00, -5.1493e+00,
         3.5332e-01,  5.2291e-01,  5.9803e+00,  4.0662e+00, -4.2361e-01,
         5.1408e+00,  2.1840e+00,  4.3019e-01,  9.2390e+00, -1.2961e+00,
         3.3133e+00,  6.8763e+00,  3.1870e+00,  5.9079e+00,  7.1879e+00,
         3.5529e+00, -5.0814e+00,  8.3206e+00,  5.8442e+00,  1.3957e+01,
         9.2200e+00,  8.3154e+00,  2.8462e+00,  7.4967e+00, -3.1369e-01,
         8.0106e+00,  3.5290e+00,  2.6430e+00,  2.1753e+00,  5.4681e+00,
         1.0434e+00,  4.4590e+00,  2.2379e+00,  2.6868e+00,  8.7966e+00,
        -3.8689e-01,  2.5630e+00,  5.2570e+00,  4.5057e+00,  2.4929e+00,
        -3.3448e-01,  2.4066e-01,  5.1192e+00,  1.2107e+01,  5.2340e+00,
        -1.0868e+00,  3.9068e+00,  1.6034e+00,  3.3359e+00,  4.4505e+00,
         3.9065e+00,  4.8776e+00,  1.3393e+01,  2.9198e+00,  3.5968e+00,
         2.9472e+00,  3.3864e+00,  6.2105e+00, -3.1747e+00,  7.0096e+00,
         2.2043e+00,  2.2660e+00,  4.6964e+00,  7.8220e-01, -1.3668e+00,
        -4.6972e+00,  8.6984e+00,  6.6832e-01,  1.2116e+01,  1.1459e+01,
        -1.6143e+00,  5.8981e+00,  3.6181e+00,  8.8860e+00,  1.9814e-01,
        -4.2476e+00,  5.4593e-01,  1.0103e+00,  4.7074e+00,  1.0846e+01,
         6.3443e+00,  8.0683e+00,  9.4812e+00,  2.5149e+00,  6.0395e+00,
        -4.1477e-01,  4.2403e+00,  4.7753e+00,  3.2651e+00, -8.8652e-01,
         6.6078e-01,  3.8812e+00,  4.8137e+00,  5.0932e+00,  2.2062e+00,
         4.3813e+00,  7.7261e+00,  5.2972e+00,  3.9683e+00,  2.9868e+00,
        -3.1202e-01,  5.8510e+00,  2.4028e+00,  3.3676e+00,  1.2410e+00,
         8.9542e+00, -2.4485e+00,  2.5508e+00, -2.1715e+00, -8.6844e+00,
         2.5331e+00, -4.3032e+00,  7.1626e+00,  2.2497e+00,  6.5137e+00,
         3.7335e+00,  5.5616e+00,  2.3892e+00,  2.7290e+00, -5.8348e+00,
        -5.4644e+00,  4.5057e+00,  8.4609e+00, -3.3843e+00,  3.1707e-01,
         1.5476e+00,  2.3913e+00, -4.4954e-02,  4.3761e+00, -8.2839e-02,
         3.6674e+00,  4.2113e+00,  3.7124e+00,  3.8790e+00,  2.5218e+00,
         7.1249e+00,  3.0214e+00,  1.9917e+00,  1.4850e+00,  7.7832e+00,
         8.3176e+00,  8.2501e+00,  8.2007e+00,  5.4913e+00,  2.8792e-01,
         2.8847e+00,  1.1983e+01,  3.8084e+00,  4.2707e+00,  3.0986e+00,
         6.0474e+00,  1.9788e+00,  4.3849e+00,  1.5986e+00,  3.6054e+00,
         3.6037e+00,  1.1726e+01,  1.0936e+01,  6.4522e+00,  1.0890e+01,
        -3.7781e+00,  6.2028e+00,  5.3935e+00,  1.0151e+01,  4.4619e+00,
        -4.0167e+00,  6.8759e+00,  3.0019e+00,  5.3817e+00,  2.5717e+00,
         1.5960e+01,  3.4484e+00,  1.4152e+00,  1.0797e+01,  3.8369e+00,
         7.4960e+00, -6.0460e-01,  5.1285e+00,  2.0571e+00,  4.0567e+00,
         4.1971e+00,  7.6146e+00,  5.3178e+00,  5.9834e+00,  7.4834e+00,
         1.9521e+00,  8.0143e+00,  3.6463e+00,  1.2679e+01, -1.1626e+00,
         2.6610e+00,  5.6225e+00,  2.6319e+00,  1.0339e+01,  1.1451e+01,
         4.4491e+00,  2.0282e+00,  8.6735e+00,  1.0155e+01,  1.1997e+01,
         4.5307e+00,  1.5682e+01,  3.4857e+00, -1.5691e-01,  4.0569e-01,
         5.9681e+00,  5.8549e-01,  6.0353e+00,  2.8781e+00,  3.9022e+00,
         3.7304e+00,  9.6522e+00,  9.3134e+00,  8.2876e+00,  2.5979e+00,
         2.0689e-02, -4.7752e+00,  2.3397e-01,  7.6722e+00,  3.0787e+00,
         6.3900e+00,  5.5710e+00,  2.3301e+00,  9.6049e+00,  8.4671e+00,
         4.2064e+00,  9.4621e+00,  9.8009e+00,  1.2833e+00,  8.7223e+00,
         4.2468e+00, -3.5599e-02,  2.3438e+00,  8.3571e+00,  2.5416e+00,
        -7.7609e-01,  1.7919e+00,  4.0481e+00,  2.7385e+00,  5.9270e+00,
         9.8819e+00,  1.9450e+00,  4.8462e+00, -9.3802e-01,  6.9600e+00,
        -1.6316e-01,  5.1814e+00, -5.7637e+00,  3.8859e+00,  7.3990e+00,
         4.5597e+00,  7.5662e+00,  6.5434e+00,  4.2004e+00,  6.0943e+00,
         6.1792e+00,  2.7396e+00,  5.3388e+00,  7.0924e+00,  1.6824e+00,
         6.1319e+00,  2.1188e+00,  1.2724e+01,  8.3875e+00,  6.9756e+00,
        -5.1477e-01,  3.1330e+00,  6.3397e+00,  4.3833e+00,  5.7131e+00,
         1.1772e+01,  1.7587e+00,  4.8179e+00,  7.9435e+00,  5.5303e+00,
         5.8392e+00,  1.1383e+01,  5.6536e+00,  3.5487e+00, -1.6380e+00,
        -4.6982e+00,  1.0899e+01,  3.5066e+00,  1.6236e+00,  7.4673e+00,
         1.0255e+00,  4.0808e+00,  3.3368e+00, -1.0236e+00,  1.2625e+01,
        -1.3387e+00,  2.3446e+00,  3.3586e+00,  6.6342e+00,  5.7768e+00,
         9.0653e+00,  8.1414e+00,  8.8771e+00,  4.4019e+00,  7.6883e+00,
         6.4577e+00,  8.3485e+00,  1.2736e+01,  4.2306e+00,  5.5936e+00,
         7.6804e+00,  2.9430e+00,  6.7353e+00,  9.8066e-01,  3.2055e+00,
         1.3476e+00,  7.5647e+00, -1.8126e+00, -2.9471e+00,  5.6616e+00,
         1.7224e+00,  8.7692e+00,  7.9790e+00,  1.0145e+01, -3.5840e+00,
         5.9221e+00,  5.7394e-01,  5.5511e+00,  2.5817e+00, -2.5728e+00,
         6.3243e+00,  1.0157e+01,  4.9175e+00,  7.2890e+00, -4.9364e-01,
         1.4279e+01,  7.1839e+00,  1.2499e+00,  5.7143e+00,  8.1935e+00,
         9.9804e+00,  8.2209e+00, -3.9507e+00,  2.8999e-01,  5.2902e+00,
         3.6858e+00,  9.8107e-01, -5.1965e+00, -2.2708e-02, -5.2105e-01,
        -2.9165e+00,  3.0669e+00,  5.7150e+00,  3.2251e+00,  8.3505e+00,
         1.2982e+00, -5.2331e+00,  6.2752e+00,  4.5034e+00,  4.2575e+00,
         8.7604e-02,  6.7349e+00,  1.8515e+00, -2.7667e-01, -4.4933e+00,
         1.0564e+01,  4.7346e+00,  5.8366e+00,  2.6844e+00,  1.0927e+00,
         3.1951e+00,  7.8895e+00,  3.4890e+00,  2.2100e+00, -6.2742e+00,
         8.3406e+00,  7.2475e+00,  6.6718e+00,  4.4655e+00,  1.1956e+01,
         8.8555e+00,  6.6663e-01,  3.4600e+00,  6.0791e+00,  2.9571e+00,
         7.9097e+00,  6.5524e+00,  5.0347e+00,  5.3920e+00,  1.1962e+01,
         9.1899e+00, -2.9796e+00, -1.5160e+00,  4.6095e+00,  1.6824e+00,
         1.1531e+01,  6.3497e+00,  7.9342e+00,  8.8920e-01,  3.9356e+00,
         2.8434e+00,  3.3933e+00,  1.5888e+00,  4.3775e+00,  6.0910e-01,
         3.5702e+00,  2.9208e+00,  3.2796e+00,  2.4769e+00, -8.8868e-01,
         2.2371e+00,  4.1302e+00,  3.4660e+00,  2.5509e+00,  5.6532e+00,
         8.5912e+00,  5.6743e-01, -1.2834e+00,  2.1839e+00,  1.5710e+00,
         6.1612e+00,  3.0325e+00, -1.9935e+00,  6.1765e-02,  7.8360e+00,
         8.4672e+00,  3.1817e+00,  7.6683e+00,  7.1307e+00,  1.0541e+01,
        -3.1593e+00,  8.7310e+00, -1.3903e+00, -2.8871e+00,  2.9730e+00,
         2.8498e+00,  6.2631e+00,  8.3476e-01,  1.2212e+00,  2.8533e+00,
         5.4020e+00, -3.2458e+00,  4.6562e+00, -1.7852e+00,  2.8612e+00,
         3.6934e+00,  5.9106e+00,  7.5514e+00,  1.0076e+01, -2.1726e+00,
         4.4809e+00,  4.4783e+00,  7.0596e+00,  2.3858e-01,  4.8439e+00,
        -2.3723e+00,  1.6071e+00,  3.6186e+00,  9.4666e+00, -4.8750e+00,
        -3.4768e-01,  1.5201e+00,  6.0265e+00, -2.7023e+00,  7.0703e-01,
         1.4886e+00,  4.6137e+00,  2.5571e+00,  9.1164e+00,  2.1813e+00,
         2.4251e+00, -5.4174e+00,  3.0370e+00,  6.7013e+00,  9.3473e+00,
         8.4479e+00,  4.4807e+00,  3.9912e+00,  5.6223e+00,  5.9864e+00,
        -5.5813e-01,  4.2693e+00,  1.0716e+01,  2.8494e+00,  7.7159e+00,
         5.1601e+00,  4.8571e-01,  5.5176e+00,  6.6519e+00,  5.1358e+00,
        -3.0026e+00, -5.1955e-01,  1.0297e+01, -8.9967e-02, -1.5106e+00,
        -1.1321e+00,  2.8333e+00,  6.1228e+00,  9.6462e+00,  2.3384e+00,
         5.3950e+00,  4.4062e+00,  1.0180e+01,  4.3470e+00,  2.7872e+00,
         1.8611e+00,  2.5134e+00,  5.7156e+00, -6.9156e-01,  4.5747e+00,
         9.7850e+00,  3.7372e+00,  1.2928e-02,  3.2591e+00,  2.5492e+00,
         6.6300e+00,  4.5103e+00,  7.5360e+00,  4.4841e+00,  2.6863e-01,
         4.1175e+00,  3.8141e+00,  8.5683e+00,  5.7411e+00, -1.8767e+00,
         3.3068e+00,  7.5902e+00,  1.7992e+00,  6.4096e+00, -1.4747e+00,
         7.5955e-01, -1.1165e+00, -8.3351e-01,  8.0473e+00,  7.0206e+00,
         2.7854e-01,  8.3425e+00,  1.2732e+01,  3.6333e+00,  3.9375e+00,
        -3.1737e+00,  5.8484e+00,  1.1025e+01,  2.4318e+00,  2.4235e+00,
         2.0651e+00,  3.0804e+00,  3.7958e-01,  1.0392e+01,  1.9037e+00,
         5.9838e+00,  3.1991e+00,  1.2407e+01,  2.3793e+00,  9.9795e+00,
         6.2982e+00,  7.5077e+00,  8.1641e+00,  6.9588e+00,  6.0153e+00,
         4.6290e+00,  5.7888e-01,  4.9759e+00,  5.1604e+00,  1.4878e+00,
         1.3879e+00, -3.3612e-03,  4.7251e+00,  1.4042e+00,  7.6185e+00,
         4.6528e+00, -5.5974e-01,  4.1485e+00,  6.0415e+00,  3.7076e-02,
        -2.8002e-01,  3.6932e+00,  3.8383e+00,  1.3709e+00,  4.6547e+00,
        -1.0284e+00, -3.8996e+00,  3.6553e+00,  6.7843e+00,  7.4686e+00,
         4.0237e+00,  7.1377e+00, -1.1221e+00,  4.1713e+00,  1.0945e+00,
         6.8784e+00,  5.7325e+00,  4.2118e+00,  3.8307e+00, -1.9557e+00,
         6.4028e+00,  1.4229e+00,  1.1147e+01,  2.1697e-01,  7.1066e+00,
         1.0422e+01,  5.8440e+00,  4.7316e+00,  6.0332e+00,  7.1857e+00,
         1.6228e+01,  8.0710e+00,  4.8227e+00, -1.9242e+00,  1.3944e+00,
         2.2093e+00,  1.1174e+01, -2.6797e+00,  4.4563e+00,  1.4998e+00,
         8.7097e+00,  4.7602e+00,  8.2786e+00,  2.3423e+00, -1.1977e+00,
        -5.4601e+00,  9.3126e+00,  6.5522e+00,  5.5415e+00,  9.4498e+00,
         8.4900e+00,  4.9120e+00,  2.9287e+00,  3.2775e+00,  7.3237e+00,
         4.5565e+00,  5.7191e+00,  6.1996e+00, -2.6642e+00,  1.6862e-01,
         5.5764e+00,  2.4458e+00,  8.3643e+00,  6.5782e-01,  6.1868e+00,
         2.0528e+00,  3.1640e+00,  1.1405e+00, -5.3776e-02,  5.9687e+00,
         9.0724e+00, -1.0499e+00,  1.7880e+00,  4.3890e+00,  4.5123e+00,
         4.6943e+00,  5.8173e+00, -5.0093e-01,  6.0371e+00,  9.3917e+00,
         3.0151e-01,  2.8057e+00,  4.6947e+00,  3.9927e+00,  5.1564e+00,
        -2.6950e+00,  4.5783e+00,  4.1164e-01,  2.1895e-01,  1.2455e+00,
         6.7267e+00,  7.8661e+00,  4.2238e+00,  7.7975e-01,  8.4252e+00,
         1.0888e+01, -1.0680e+00, -1.2206e+00,  6.3811e+00,  4.1526e+00,
         5.5986e-01,  1.0509e+00,  1.7467e+00,  3.2737e+00,  4.2207e+00,
        -6.5776e-01, -1.2718e+00,  1.4231e+00, -5.9610e-01,  7.2578e+00,
         9.9314e+00,  2.6374e+00, -2.4831e-01,  2.6269e+00, -9.8466e-01,
         4.9932e-01, -2.0461e+00,  6.6173e+00,  1.8805e+00, -3.0255e+00,
         1.1837e+01,  2.4930e+00,  3.3906e+00,  1.1015e+01,  9.3018e+00])