是否有INDArray的方法来执行两个INDArray（大小不同）的加法运算？

Question

想象两个INDArray：

• a1.shape（）= [2，3]并且a1充满1;

• a2.shape（）= [1，2]并且a2充满2。

我想在它们之间进行加法，例如：

?> result = a1.add(0, a2)
?> print(result)
[[3, 3, 1], [1, 1, 1]]
?> result = a1.add(1, a2)
?> print(result)
[[1, 1, 1], [3, 3, 1]]
?> result = a1.add(1, 1, a2)
?> print(result)
[[1, 1, 1], [1, 3, 3]]

我尝试首先选择一个子数组：

?> subarray = a1.get(NDArrayIndex.interval(0, 1), NDArrayIndex.interval(0, 2))
?> print(subarray)
[1, 1]

执行添加：

?> subarray = subarray.add(a2)
?> print(subarray)
[3, 3]

但是我不知道如何在合适的位置将'subarray'插入'a1'中...

注意：为了便于说明，我简化了问题。正在处理的数组是4D数组。

Answer 1

我刚刚在文档中找到了以下功能：

INDArray put(INDArrayIndex[] indices, INDArray element)

这可以用来解决这个问题：

a1.put(
  new INDArrayIndex[] {NDArrayIndex.interval(0, 1), NDArrayIndex.interval(0, 2)},
  subarray
);

实际上是NDArrayIndex实现了INDArrayIndex接口。