我正尝试在我的损失函数中使用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
以将?
批次大小作为其主要尺寸?