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

时间:2020-09-09 22:30:31

标签: javascript tensorflow tensorflow.js

让我澄清这个问题。我有一个张量,只是为了简单起见,为零。形状为[3, 3]的2D张量,称为a

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

我还有另一个张量,形状为[2, 2]的2D张量,称其为b,还有它的唯一张量:

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

我想将它们加在一起,但是它们的形状不匹配,我也不能广播它,但这不是目标。我想要以下结果:

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

如果有一种方法,现在将其称为insert,则必须指定要插入较小张量的位置(就像在切片时定义起始位置一样)。看起来像这样:

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

我在考虑pad可能会产生相似的结果,但是如果出现上述情况,我可能会错过,那会更好。

1 个答案:

答案 0 :(得分:1)

通过tf.pad,可以按照以下步骤进行操作

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