是否有 TensorFlow.js 方法可以在特定位置将张量添加到另一个张量？

Question

让我把问题说清楚。我有一个张量，只是为了简单起见，带有零。形状为 [3, 3] 的二维张量，称其为 a:

[[0,0,0],
 [0,0,0],
 [0,0,0]]

我还有另一个张量，一个二维张量，形状为 [2, 2]，称之为 b，它只有一个：

[[1,1],
 [1,1]]

我想把它们加在一起，但是它们的形状不匹配，我也不能播放，但这不是目标。我想要的结果是：

[[0,0,0],
 [0,1,1],
 [0,1,1]]

如果有方法，暂时称之为 insert，我必须指定要插入较小张量的位置（就像切片时一样，定义起始位置） .它看起来像这样：

a.insert(b, [1, 1])

我在想pad可以产生类似的结果，但如果有像上面那样的东西，我可能会错过，那就更好了。

Answer 1

使用tf.pad，可以按如下方式进行

padded = tf.ones([2, 2]).pad([[1, 0], [1, 0]])

Is there a TensorFlow.js method to add a tensor to another at a specific location?