luyang00 / pytorch-extension Goto Github PK
View Code? Open in Web Editor NEWThis project forked from sniklaus/pytorch-extension
an example of a CUDA extension for PyTorch which computes the Hadamard product of two tensors
License: GNU General Public License v3.0