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

posted @ 2024-07-29 00:10  zwlwf  阅读(17)  评论(0编辑  收藏  举报