GitHub repository: sniklaus/pytorch-extension
Primary language: Python
Created: 2017-07-01T04:49:16Z
Description: an example of a CUDA extension for PyTorch which computes the Hadamard product of two tensors