(tensorflow)tf.tensor_scatter_add用于具有未知batch_size的两个张量

时间:2019-04-10 15:07:37

标签: python tensorflow tensor

我正尝试在我的损失函数中使用tf.tensor_scatter_add,如下所示: -张量t,大小:(?,256,256) -形状为ndarray ind(76,) -形状为(?,76,256)的张量updates

我尝试tf.tensor_scatter_add(t, ind, updates),但这不起作用。我收到此错误:

  

ValueError:output.shape = [?, 256,256]的内部-73尺寸必须与updates.shape = [?, 76,256]的内部3个尺寸相匹配:形状必须等于等级,但对于' loss / conv2d_15_loss / TensorScatterAdd'(op:'TensorScatterAdd'),输入形状为[?,256,256],[76],[?, 76,256]。

如何重塑ind以将?批次大小作为其主要尺寸?

0 个答案:

没有答案