Failing that you could implement a funtion in c with the cuda library to accept both the gpumat cuda pointer and the torch tensor cuda pointer ...
確定! 回上一頁