wwhio / pytorch-extension

an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors

Geek Repo

Github PK Tool