Я пытаюсь использовать tf.tensor_scatter_add
в моей функции потерь следующим образом: - тензор t
размера: (?, 256, 256) - ndarray ind
формы (76,) - тензор updates
формы (?, 76, 256)
Я пытаюсь tf.tensor_scatter_add(t, ind, updates)
, но это не работает.Я получаю эту ошибку:
ValueError: Внутренние -73 измерения output.shape = [?, 256,256] должны совпадать с внутренними 3 измерениями updates.shape = [?, 76,256]: формы должныбыть равным рангу, но 0 и 3 для 'loss / conv2d_15_loss / TensorScatterAdd' (op: 'TensorScatterAdd') с входными формами: [?, 256,256], [76], [?, 76,256].
Как изменить форму ind
, чтобы включить размер партии ?
в качестве начального размера?