文件名称：pytorch-extension:一个使用CuPy的PyTorch的CUDA扩展示例，该示例计算两个张量的Hadamard乘积
文件大小：15KB
文件格式：ZIP
更新时间：2024-05-23 20:03:08
python deep-learning cuda pytorch cupy
火炬延伸 这是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
【文件预览】：
pytorch-extension-master
----README.md(1KB)
----LICENSE(34KB)
----requirements.txt(24B)
----hadamard.py(3KB)
----run.py(1KB)