tensorflow потребляет слишком много памяти при использовании tf.map_fn

#python-3.x #tensorflow Вопрос: У меня есть следующий код: @tf.function def c(self, point_instances, instances, alpha): def inner_comp(j): print(j.shape) print(tf.transpose(instances).shape) print(alpha.shape) print(tf.linalg.matvec(tf.transpose(instances), alpha).shape) tmp = tf.tensordot(j, tf.linalg.matvec(tf.transpose(instances), alpha), 1) print(tmp.shape) return tmp…