pytorch-extension:一个使用CuPy的PyTorch的CUDA扩展示例，该示例计算两个张量的Hadamard乘积-源码

火炬延伸 这是PyTorch的CUDA扩展的示例，该扩展使用CuPy计算两个张量的Hadamard乘积。 有关也使用CuPy的更高级的PyTorch扩展，请参阅： : 设置 确保安装CuPy，这可以使用pip install cupy或使用CuPy信息库中概述的提供的之一来完成。 用法 无需单独的构建过程，只需运行python run.py进行测试。 下面还显示了如何使用示例扩展的最小示例。 import torch import hadamard class Network ( torch . nn . Module ): def __init__ ( self ): super ( Network , self ). __init__ () # end def forward ( self , input1 , input2 ): return hadamard