Как сделать нарезанное назначение в Keras (используя TensorFlow в качестве бэкэнда)

Question

Например,

a = Input(...)
b = keras.layers.Conv2D(...)(a)
c = keras.backend.zeros(...)

c[...].assign(b[...])

Поскольку он используется до компиляции модели, когда я пытаюсь использовать функцию assign() (это функция TensorFlow), возникает ошибка: 'Tensor'Объект не имеет атрибута «назначить».

Вероятно, это связано с тем, что перед компиляцией модели первый dim переменной равен None.Итак, есть ли способ сделать нарезанное назначение?

Answer 1

Как правило, тензоры TensorFlow не присваиваются. Однако, согласно официальной функции docs , tf.assign() работает только с изменяемыми тензорами, которые должны быть из узла Variable. Итак, код ниже работает. Остальное будет сильно зависеть от вашего конкретного случая.

var1 = tf.keras.backend.zeros(1,1)
var2 = var1[0].assign(1)

sess = tf.Session()
sess.run(tf.global_variables_initializer())
print(sess.run(var2))

Вы можете найти этот ответ полезным Как сделать назначение срезов в Tensorflow