Optimization TensorFlow中约束优化方法的实现?

Optimization TensorFlow中约束优化方法的实现?,optimization,tensorflow,Optimization,Tensorflow,我正在寻找类似scipy.optimize中存在的l_bfgs_b或tnc方法。有人见过这些工具的实现吗?您可以看看。这使您可以直接在TensorFlow模型上使用Scipy优化器。Oh wow。这些是在GPU上进行优化,还是在外部调用scipy​?该模型本身当然可以在GPU上运行,但不幸的是,为了优化步骤本身,将有相当多的数据来回复制到CPU(即,它实际上使用的是Scipy优化器,而不是将其插入TensorFlow模型)。对于某些模型(大量计算,而不是太多的参数),它可以正常工作。对于在GPU

我正在寻找类似scipy.optimize中存在的l_bfgs_b或tnc方法。有人见过这些工具的实现吗?

您可以看看。这使您可以直接在TensorFlow模型上使用Scipy优化器。

Oh wow。这些是在GPU上进行优化,还是在外部调用scipy​?该模型本身当然可以在GPU上运行,但不幸的是,为了优化步骤本身,将有相当多的数据来回复制到CPU(即,它实际上使用的是Scipy优化器,而不是将其插入TensorFlow模型)。对于某些模型(大量计算,而不是太多的参数),它可以正常工作。对于在GPU上运行几毫秒的模型,它会减慢速度。@AllenLavoie如何解决组合约束问题,例如
a+b+c=1
equalities=[a+b+c-1.]
?假设
a
b
c
,是
张量
s。