gpu - 以矢量方式计算期权收益（无循环）

Question

大家好

我想用 Nd4J 为普通期权定价。

所以我希望用 Nd4J 实现操作

最大值（0，X[i]-K）=> Y[i]

X 输入 INDArray K 浮点数 Y 输出 INDArray

有人知道如何进行吗？

提前感谢您给我的时间

Answer 1

我正在回答我自己

INDArray Y=Transforms.max(X.add(-K), 0f);