将函数元素应用于Pytorch CUDA Tensor的并行方法

Question

假设我有一个火炬CUDA张量，我想应用像sin()这样的函数，但我已经明确定义了函数F。如何使用并行计算在Pytorch中应用F。

Answer 1

我认为目前，无法在CUDA-Tensor上显式并行化函数。一种可能的解决方案是，您可以定义一个函数，例如非线性激活函数。所以你可以通过网络和你的功能来提供它。

缺点是，它可能不起作用，因为你必须定义一个CUDA函数并且必须重新编译pytorch。