luyang00 / torch-extension

an example of a CUDA extension for Torch which computes the Hadamard product of two tensors

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

luyang00/torch-extension Watchers