假设我有一个火炬CUDA张量,我想应用一些函数,sin()但我明确定义了该函数F.如何F在Pytorch中使用并行计算.
sin()
F
gpu torch pytorch tensor
gpu ×1
pytorch ×1
tensor ×1
torch ×1