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

Question

让我把问题说清楚。我有一个张量，只是为了保持简单，用零。具有 shape 的 2D 张量[3, 3]，称之为a：

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

我还有另一个张量，一个具有 shape 的 2D 张量[2, 2]，称之为b，它只是一个：

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

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

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

如果有一个方法，insert现在调用它，我必须指定我想要插入较小张量的位置（就像切片时定义起始位置一样）。它看起来像这样：

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

我在考虑pad哪个可以产生类似的结果，但是如果有类似上面的东西，我可能会错过什么，那就更好了。

标签： javascripttensorflowtensorflow.js

Answer 1

，tf.pad它可以如下完成

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