java - 有没有一种方便的方法来按元素反转 Nd4j 数组？

Question

例如，我有 x = [1, 2, 3]。我想要 y = 1/x = [1, 1/2, 1/3]。

理想情况下，我想要一个与nd4j 文档中的以下行相同的函数，仅适用于乘法逆而不是双曲正切。

 INDArray tanh = Nd4j.getExecutioner().execAndReturn(new Tanh(myArr))

如果没有这样的函数，是否有比使用 for 循环和单独反转每个值更有效或至少更清洁的方法来对数组执行自定义元素操作？

Answer 1

您可以使用Pow转换操作 with-1.0作为指数参数：

INDArray inv = Nd4j.getExecutioner().execAndReturn(new Pow(myArr, -1.0));