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([ 3.8289e+00,  1.8653e+00,  4.4250e+00,  2.8341e+00,  1.1564e+01,
         7.3641e+00,  1.5435e+00,  6.7380e+00,  6.9383e+00,  3.6823e+00,
        -2.6328e+00,  6.9275e+00,  1.8300e+00,  8.1740e-01,  6.9061e-01,
         4.9822e+00,  2.4043e+00,  6.2444e+00,  7.5422e+00,  3.8553e+00,
         6.5205e+00, -3.1938e+00,  3.3807e+00,  1.1596e+00,  1.2209e+01,
         1.1756e+00, -1.7697e+00,  1.9882e+00,  4.4324e+00,  3.9200e+00,
         1.7437e+00,  9.1596e+00,  3.0908e-01, -4.2704e-01,  4.6867e+00,
         4.6546e+00,  3.5735e+00,  4.5154e+00,  2.4847e+00,  8.0081e+00,
         2.2720e-01,  7.3791e+00,  4.9551e+00,  5.7723e-01,  7.7501e+00,
         7.9209e+00,  7.7051e+00,  3.2006e+00,  6.4874e+00, -1.3575e+00,
         1.1146e+01,  8.8385e+00,  9.6623e+00,  2.4354e+00,  5.0900e+00,
         7.0097e+00,  5.0592e+00,  4.7131e+00,  7.7792e+00,  1.2882e+01,
         7.7923e+00, -5.9042e+00,  1.7792e+00,  6.9348e+00, -7.3768e-01,
         1.9122e+01,  1.5661e+00,  7.5634e+00,  1.1867e+00, -5.5177e+00,
         1.1797e+00,  1.7348e+00,  3.6367e+00,  4.7972e+00,  1.0702e+01,
         7.1279e+00,  1.1575e+01, -2.3692e-01,  5.6573e+00,  1.4267e+00,
         7.4770e+00,  4.2573e+00,  2.8344e+00,  1.2884e+01,  3.7619e+00,
         2.1157e+00, -5.0405e+00,  4.4148e+00, -2.8004e-01,  5.5866e+00,
         3.6628e+00,  5.5358e+00,  5.2130e+00,  4.8571e+00,  5.8654e+00,
         1.0976e+01,  4.0743e+00,  6.7796e+00,  2.4415e+00, -8.3995e-01,
         8.6687e+00,  8.0857e+00,  5.3988e+00,  1.1404e-01,  5.4502e+00,
         3.2440e+00,  3.2983e+00,  3.4085e+00,  2.3573e+00, -4.6592e-01,
         1.5888e+00,  4.6445e+00, -8.8332e-01,  6.6872e-01,  8.5452e+00,
         4.7300e+00,  2.3966e+00,  2.2871e+00,  1.1948e+00,  3.3248e+00,
         6.8790e+00,  7.2492e+00,  4.9568e+00,  3.6364e+00,  6.5738e+00,
         2.2161e+00,  1.5375e+00,  7.8257e+00, -2.3700e+00,  7.8435e+00,
         4.5652e+00,  4.9748e+00,  3.0418e+00,  9.7638e+00,  6.4300e+00,
        -2.8424e+00,  2.8226e+00,  3.6797e+00,  4.3778e+00,  5.2391e+00,
         7.8702e+00, -1.6779e+00, -3.4724e-01,  9.1386e+00,  1.0262e+01,
         2.4466e+00,  5.5900e+00,  5.4683e+00, -1.0685e+00,  1.1479e+01,
         2.7841e+00,  2.3385e+00,  1.4288e+00,  9.6571e+00,  7.4572e+00,
         9.0377e-01,  6.2744e+00,  1.5167e+00,  1.3240e+00,  7.1366e+00,
         9.5067e-01,  5.8499e+00,  5.5584e+00,  2.3026e+00,  6.0253e+00,
        -9.3205e-01,  4.6109e+00,  1.1302e+01,  2.7105e+00,  4.8870e+00,
         4.8759e+00,  6.8026e+00,  2.2313e+00,  2.4255e+00,  6.0482e+00,
         1.2723e+01,  8.7581e-01, -2.2944e+00,  4.2863e+00,  2.9022e+00,
         2.4998e+00, -1.2041e+00,  2.6947e+00,  4.8205e+00,  4.6222e+00,
         7.8177e+00,  5.8179e+00,  1.8793e+00,  8.2589e+00,  5.7179e+00,
         1.1391e-01,  3.4999e+00,  5.1839e+00,  2.3552e+00, -4.0115e+00,
         7.6812e+00,  3.5088e+00,  2.1880e+00,  7.1080e+00,  1.0695e+01,
         1.6472e+00,  6.1965e-01,  4.2800e+00,  4.1884e+00,  5.2356e+00,
         2.9275e+00, -1.2910e+00,  9.0235e+00,  7.7314e-01,  7.0032e+00,
         5.2167e+00,  8.0027e+00,  1.8234e-01,  1.1820e+01, -1.2468e+00,
         5.4537e+00,  7.0476e+00, -2.7688e+00,  3.1378e+00,  4.0360e+00,
         1.5146e+00,  3.9713e+00,  4.2306e+00, -2.9371e+00,  3.0600e+00,
         5.3721e+00,  1.5765e+00,  7.8596e-01,  3.7244e+00,  5.9734e-01,
         9.6478e+00,  2.8681e+00,  2.6969e+00,  1.3551e+01,  3.1747e-01,
        -7.2820e-01,  8.9842e+00,  6.9841e+00,  7.6227e+00,  4.3766e+00,
         5.7428e+00,  2.6936e+00,  7.4957e+00,  6.0872e+00,  7.7484e+00,
         1.4747e+00,  1.2116e+01,  6.0066e+00, -3.4782e+00,  1.3750e+01,
        -1.2816e+00,  8.4102e+00,  3.8111e+00,  8.1302e+00,  1.2970e+01,
         5.6738e+00, -3.8072e-01, -4.9330e-01,  6.6394e+00,  3.2384e+00,
         7.8299e+00, -5.6673e-02, -3.0377e-01,  1.2218e+01,  4.4609e+00,
         8.8645e+00,  8.3323e+00,  8.0787e+00,  3.6171e+00,  5.6150e+00,
         8.6544e+00,  1.3775e+00,  5.0366e+00,  1.1919e+01,  1.3356e+01,
         5.3615e+00,  4.7618e+00,  1.5006e+00,  8.3289e+00,  4.3708e+00,
         7.5786e+00,  3.4935e+00, -6.0602e-01, -1.8066e+00, -1.7758e+00,
        -1.3188e+00,  5.8643e+00, -2.0638e+00,  7.6129e+00,  1.0461e+00,
         4.2968e+00,  2.1600e+00,  6.1272e-01,  8.2493e+00,  4.2148e+00,
         5.3506e+00,  8.0855e+00, -2.0102e-01,  3.8803e+00,  2.0895e+00,
         9.6980e+00,  3.1011e+00, -9.4717e+00,  8.2297e+00,  3.6907e+00,
         8.4494e+00,  6.0321e+00, -2.3443e+00, -2.6188e+00,  3.6752e+00,
         1.9861e+00,  1.3672e+01,  4.8282e+00,  2.4781e+00,  8.4451e+00,
         6.1945e+00,  8.2838e-01,  6.0372e+00, -4.1709e+00,  5.4536e+00,
         1.6873e+00,  1.8109e+00,  1.1404e-01, -5.3159e-01,  2.0842e+00,
         8.4753e+00, -2.0711e+00,  1.0286e+01,  4.1473e+00,  5.0697e+00,
         2.4708e+00,  8.7008e+00,  3.2017e-01,  4.3103e+00,  7.1403e+00,
         3.9752e+00,  5.2669e+00,  4.0822e+00, -3.3117e+00,  4.4475e+00,
         1.0138e+01, -2.7662e+00,  4.2820e+00,  7.3187e+00,  5.3829e+00,
        -7.2249e-01, -4.2877e+00, -1.6853e+00, -4.2105e+00,  2.7645e+00,
         6.4053e+00,  7.9919e+00,  8.4311e+00,  7.6368e+00,  7.7862e+00,
        -8.3651e-01,  7.3818e+00,  1.0596e+01,  2.3163e-01,  6.5328e+00,
         5.5052e+00,  3.0574e+00,  4.7660e+00,  5.6557e+00,  9.4770e+00,
         4.6035e-01,  8.9908e+00,  4.8521e+00,  5.0184e+00,  6.5705e+00,
         7.6659e+00,  5.0835e+00,  8.0364e+00,  1.7811e+00,  2.0721e+00,
         8.4170e+00,  4.6298e-01,  6.4001e+00,  7.2656e+00,  2.0633e+00,
         1.6492e+00,  4.9831e+00,  5.0862e+00, -1.8662e+00,  3.8825e+00,
         5.4857e+00,  1.5146e+00,  4.2770e+00,  2.0633e+00,  4.1265e+00,
         3.9371e+00,  5.5918e+00,  6.3636e+00, -4.7308e+00,  3.7110e+00,
        -7.4326e+00,  7.6712e+00, -3.6654e-01,  4.1289e-02,  5.4796e+00,
         5.9173e+00,  9.2184e+00,  9.1571e+00,  5.8816e+00,  1.0704e+01,
         4.7000e+00,  3.5579e+00,  4.4581e+00,  8.7599e+00,  1.8078e+00,
         2.0042e+00,  4.5345e+00, -8.7786e-01,  2.5688e-01,  9.5640e+00,
         1.9184e+00,  7.6941e+00,  6.1249e+00,  7.5320e+00,  1.7006e+00,
         5.9199e+00,  5.2746e+00,  1.1669e+01,  3.7318e-01,  3.1319e+00,
         5.8243e+00,  9.5687e+00,  5.1392e+00,  5.4209e-01,  3.7941e+00,
         3.1288e+00,  8.5105e+00, -2.0753e+00,  2.5900e+00,  9.1257e-01,
         5.4238e-01,  2.1580e+00, -2.5581e+00,  2.5446e+00,  4.7963e+00,
         6.9528e+00,  4.3912e+00,  3.6268e+00,  1.5549e+00,  9.1948e+00,
         2.7068e+00,  4.2625e+00,  1.1407e+01,  3.0765e+00,  9.0124e+00,
         2.7686e+00,  2.1499e-01,  1.5809e+00,  7.8818e+00,  7.5120e+00,
         4.8167e+00,  9.5289e+00, -3.3075e+00,  2.9103e+00,  8.3969e+00,
         6.6481e+00,  3.9388e+00,  3.1343e+00,  5.5203e+00,  6.6929e+00,
         9.8499e+00, -1.3584e-01,  3.3576e+00,  3.1183e+00,  4.4397e+00,
         6.9191e+00, -2.2791e+00,  1.4991e+00,  1.0491e+01, -1.3689e+00,
         9.1329e+00,  4.8045e-01,  5.5773e+00,  1.2867e+01, -3.8797e-01,
         7.7819e+00,  7.9893e-01,  1.0155e+01,  8.7166e-04,  4.4004e+00,
        -4.4009e+00,  3.2950e+00,  4.0176e+00,  3.8223e+00,  8.5731e+00,
         5.9550e+00,  3.1220e+00,  7.7115e+00,  4.3224e+00,  4.5343e+00,
         3.1382e+00, -3.9196e+00,  7.0224e+00,  5.2506e+00,  6.4250e+00,
         3.4765e+00,  3.4722e+00,  5.7295e+00,  1.5559e+00,  1.2988e+00,
         9.0895e+00,  7.1141e+00, -7.3086e-01,  6.2003e+00,  7.5467e+00,
         6.0121e+00,  7.2820e+00,  9.6407e+00,  8.4240e+00,  4.6587e+00,
         7.8066e+00, -1.5918e+00, -4.9625e+00,  6.9637e+00,  7.2925e+00,
         5.8567e+00,  2.1257e+00, -1.0569e+00,  8.1814e-01,  4.9409e+00,
         5.7406e+00,  1.6971e+00,  2.2449e+00,  5.3392e+00,  3.5607e+00,
        -1.6418e+00,  1.6085e+00,  5.6341e+00,  8.9043e+00, -3.5922e+00,
        -1.0611e+00,  2.0893e+00,  4.1510e-01, -1.8799e+00,  4.2587e+00,
         8.2358e+00,  2.8391e+00,  1.0134e+00,  3.9331e+00,  5.0003e+00,
         2.5495e+00, -1.5060e+00,  1.9288e+00,  6.2388e+00, -1.1439e-01,
         1.0993e+01,  5.0739e-01,  1.8038e-01,  1.2867e+00,  8.4176e+00,
         3.7053e+00,  6.9974e+00, -6.9006e-01, -5.5613e+00,  3.5913e+00,
         6.9323e+00,  2.0603e+00,  5.5289e+00,  6.9384e+00,  9.1396e+00,
         1.0627e+01, -2.0577e+00,  1.0101e+01,  3.8708e+00, -1.0628e+00,
        -1.1778e+00,  2.9096e+00,  4.6555e-01,  3.0348e+00,  5.2168e+00,
         1.2719e+01,  1.0204e+01,  1.0531e+01, -8.5914e-01,  1.5490e-01,
         4.0697e+00,  8.4176e+00,  4.6673e-01, -3.1970e-01,  9.8603e+00,
         9.9455e+00,  5.5101e+00,  8.9326e+00,  6.8982e+00,  1.7056e+00,
         1.4089e+01,  6.8283e+00,  5.8335e-01,  2.3766e+00,  6.1915e+00,
        -1.8804e-01,  3.1377e-01,  3.8521e+00,  2.4662e+00,  4.5409e+00,
        -3.3400e+00,  5.1212e+00,  1.5552e+00,  2.8716e+00,  6.3285e+00,
         9.8643e+00,  1.2161e+00,  1.1429e+00, -1.6547e-01, -6.7255e-01,
         3.7712e+00, -6.3492e-02,  6.3928e+00,  3.2933e+00,  4.2349e+00,
         5.3054e+00,  6.2617e+00,  4.5948e+00,  1.7616e+00,  2.5714e+00,
         1.3368e+01, -3.9517e-01,  7.5030e+00,  1.2207e+00,  1.9582e+00,
         8.7542e-01, -2.2549e+00,  7.5660e+00,  9.4256e+00,  2.2962e+00,
         8.0365e+00,  3.0468e+00,  3.8175e+00,  1.1239e+01,  2.8870e+00,
         4.4291e+00,  1.2489e+00,  1.2754e+01,  5.0992e+00, -2.2847e+00,
         5.5897e+00,  8.3802e+00,  4.1270e+00,  3.3378e+00,  7.1129e+00,
         7.5632e+00,  3.2792e+00,  4.0565e+00,  3.8526e+00,  6.4436e+00,
         6.1711e+00,  3.1794e+00,  4.9637e+00,  5.0939e+00,  4.6548e+00,
         8.8568e+00,  6.9357e+00,  1.7641e+00,  4.8324e+00,  3.1636e-01,
         5.9838e+00,  7.0807e+00,  8.9091e+00,  8.3587e+00,  7.5595e+00,
         2.6727e+00,  2.5542e+00,  3.7933e+00,  3.9882e+00,  1.0128e+01,
         5.5575e+00,  6.9742e+00,  2.7521e+00,  6.5831e-01,  3.5171e+00,
         3.8913e+00, -8.4814e+00,  7.6499e+00,  1.1879e+01,  6.4842e+00,
         1.8110e+00,  6.6068e+00,  5.9124e+00,  4.6266e+00,  9.5506e+00,
         1.2613e+00,  6.8491e+00,  1.3809e+00,  4.7300e+00, -3.6017e+00,
         1.2959e+01,  9.3701e+00,  4.3727e+00,  5.4872e+00,  1.3131e+01,
         8.8655e+00,  4.8467e+00,  8.0729e+00,  7.9140e+00,  3.5976e+00,
         4.3385e+00,  1.1025e+00,  8.1103e+00,  2.4768e+00,  1.0532e+01,
         3.5845e+00,  4.3771e+00,  2.0906e+00,  1.6844e+00,  7.0347e+00,
         3.8315e-01,  2.4739e+00,  6.1010e+00,  8.5459e+00,  2.9301e+00,
         5.7037e+00,  8.9139e+00,  7.9468e+00,  4.3261e+00,  1.9833e+00,
         7.2611e+00,  4.3622e+00,  6.9264e+00,  2.6488e+00,  5.6285e+00,
         1.1153e+01,  5.2541e+00,  8.8591e+00, -3.5490e+00,  6.9455e+00,
         1.1980e+01,  1.0672e+01,  3.5957e+00, -1.1930e+00,  9.0231e+00,
         7.5161e-01,  4.0272e+00,  6.1486e+00,  6.1534e+00,  1.0285e+00,
         5.1413e+00,  1.0769e+00,  2.4731e+00,  5.6649e+00,  9.8085e-01,
        -8.8830e+00,  4.2775e+00,  6.6365e+00,  6.4650e+00,  6.1282e+00,
         1.3804e-01, -1.2350e+00,  6.3038e+00,  3.4925e+00,  8.9582e+00,
        -1.9709e+00,  6.6648e+00,  6.4347e+00,  1.0972e+00, -1.8645e+00,
         4.4147e-02,  2.0156e+00,  8.1496e+00,  1.8124e-01,  1.7077e+00,
         2.5548e+00, -1.1706e+00,  1.0988e+00,  6.5513e+00, -1.0282e+00,
         4.5195e+00, -1.1004e-02,  5.0867e+00,  9.9207e+00,  8.3251e+00,
         4.2440e-01, -2.1175e+00,  3.1548e+00,  6.9519e+00, -1.6205e+00,
         3.9437e+00,  3.8148e+00,  3.8933e+00,  1.4318e+01,  3.3332e+00,
        -2.4222e+00,  4.4977e+00,  6.9984e+00,  3.9511e+00,  1.2539e+00,
         4.1910e+00,  7.3932e+00,  1.0301e+01, -1.4558e+00,  6.5002e+00,
         1.2915e+01,  6.2186e+00,  4.8572e+00,  9.8152e+00,  9.3573e+00,
         8.9701e-01,  3.6523e+00,  6.3738e+00,  3.3181e+00,  2.5656e+00,
         1.6625e+00,  3.2002e+00,  9.0365e-01,  6.6642e+00,  4.1564e+00,
         4.7874e+00,  1.4947e+00,  1.1551e+01,  5.5309e+00,  1.1520e+00,
         1.5544e+00, -2.9918e+00,  2.0256e+00,  6.1313e+00,  8.7914e+00,
         6.8344e+00,  5.2083e+00,  1.7137e+00,  3.5319e+00,  7.2565e+00,
         1.6200e+00,  5.5823e+00,  8.0517e+00,  6.4811e+00,  8.8925e+00,
         7.4128e+00,  9.2561e+00,  6.6653e+00,  1.0270e+01,  6.6815e+00,
         4.2938e+00,  6.0358e+00,  2.3187e+00,  1.3751e+00,  4.5787e+00,
         2.9311e+00,  2.3381e+00,  5.0465e+00,  9.8912e+00,  5.5715e+00,
        -1.7147e+00,  5.3842e-01,  3.3892e+00,  1.1349e+01,  5.8797e+00,
         1.7608e+00,  5.3702e+00,  7.5625e+00,  1.9026e+00,  1.3699e+00,
         5.8522e+00,  6.4909e+00,  8.9648e+00,  1.1222e+01,  7.7006e+00,
         1.0567e+01,  5.9769e+00,  5.6493e+00,  5.1241e+00,  4.0145e+00,
        -3.7737e+00,  2.8703e-01, -1.2362e+00,  3.3998e+00,  1.1722e+00,
        -1.2771e+00,  2.8741e+00,  2.0854e+00,  5.1363e+00,  8.2821e+00,
         7.0018e+00,  5.0151e+00,  4.6137e+00, -1.2588e+00,  3.0449e+00,
        -2.3575e+00,  3.6337e+00,  1.5381e+00, -1.5207e+00,  1.3990e+00,
         5.8938e+00,  1.1346e+00,  4.1362e+00, -4.7440e+00,  1.6453e+00,
         2.7412e+00,  3.5828e+00,  2.1743e+00, -3.2341e-01, -9.9993e-01,
         3.9801e+00, -1.4643e+00,  9.1779e+00,  3.6375e+00, -1.1146e+00,
        -2.8642e+00,  8.0828e+00,  5.0478e+00,  5.3239e+00,  1.1181e+01,
         3.2359e+00,  2.8007e+00,  7.3260e+00,  5.9000e+00,  6.0517e+00,
         6.9218e+00,  7.4215e+00,  7.4120e+00,  6.5452e+00, -8.5889e-01,
         7.2897e+00,  1.0047e+01,  7.1521e-01,  4.0624e+00,  3.3427e+00,
         9.3454e+00,  5.4568e+00,  5.3026e+00,  3.4962e+00, -3.0946e+00,
         1.9715e+00,  2.8674e+00, -1.7734e+00,  4.0342e+00,  2.0209e+00,
         4.7403e+00,  2.6829e+00,  1.1420e+01,  3.4371e+00,  3.7435e+00,
         1.1796e+00,  7.6579e-01,  8.6477e+00,  1.5210e+00,  1.2748e+01,
         9.6851e+00,  7.3981e+00,  9.4490e-01,  9.0394e-02,  5.5727e+00,
         1.0716e+01,  1.7453e+00,  6.3853e+00,  1.3713e+00,  1.4342e+00,
         1.9153e+00,  1.1119e+00,  4.5670e+00,  3.5537e+00,  1.1820e+01,
         4.9853e+00, -1.7545e+00,  4.3953e+00,  3.8103e+00,  9.0283e+00,
         6.8963e+00,  1.8721e+00, -1.4095e+00,  2.0476e+00,  6.4138e+00,
         5.9776e+00,  4.0259e+00,  2.4210e+00,  4.5104e+00,  3.5442e+00,
         5.1537e+00,  1.0599e+01,  7.5275e+00,  3.2990e+00,  1.3014e+00,
        -8.3999e-01, -3.8502e-02,  4.3077e+00,  5.5806e+00,  4.9355e+00,
         1.9702e+00,  1.8324e+00,  5.6313e+00,  1.7918e+00,  4.7573e+00,
         3.9994e+00,  5.4907e+00,  5.3107e+00, -2.8917e+00,  1.0464e+01,
         4.3235e+00,  8.1131e+00,  5.4436e+00,  1.3441e+00,  1.1978e+01,
        -1.9168e-01,  6.2926e+00,  3.9277e+00,  5.7052e+00,  4.5181e+00])