Я хочу объединить некоторые тензоры, которые я получил от объекта в TensorFlow 2, есть ли какой-нибудь элегантный и дифференцируемый способ добиться этого? Вместо использования списка python и для l oop (код показан ниже), или, может быть, есть какая-либо операция / способ, подобный tf.add_n, но объединить тензоры, а не добавить его?
Zps = []
for i in range(len(self.objs)):
obj = self.objs[i]
Zps.append(obj.Zp)
Z_mc = tf.concat(Zps,0)
, когда Я использовал приведенный выше код и использовал scipy.optimize, используя gpflow, это, кажется, не дифференцируется. nb: я использовал tf2 и gpflow2