(Tensorflow) tf.tensor_scatter_add для двух тензоров с неизвестным размером batch_size - PullRequest
0 голосов
/ 10 апреля 2019

Я пытаюсь использовать 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, чтобы включить размер партии ? в качестве начального размера?

...