让我澄清这个问题。我有一个张量,只是为了简单起见,为零。形状为[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
可能会产生相似的结果,但是如果出现上述情况,我可能会错过,那会更好。
答案 0 :(得分:1)
通过tf.pad
,可以按照以下步骤进行操作
padded = tf.ones([2, 2]).pad([[1, 0], [1, 0]])