torch中的矩阵乘法
torch中的矩阵乘法
torch.matmul应该是打包的函数,/home/zwl/software/pytorch/torch/csrc/autograd/generated/python_torch_functions_0.cpp:4865
static PyObject * THPVariable_matmul(PyObject* self_, PyObject* args, PyObject* kwargs)
这个文件是编译过程中自动生成的,初始的pytorch包中没有。
pytorch编译,
set USE_CUDA=0
set DEBUG=1
set USE_XPU=0
python setup.py develop