pytorch-extension

This is a CUDA extension for PyTorch which computes the Hadamard product of two tensors.