pytorch-extension
an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
117 stars
6 watching
14 forks
Language: Python
last commit: 7 months ago cudacupydeep-learningpythonpytorch
an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors