Tensorflow tf.batch_matrix_diag（无属性错误）

Question

一台机器，我有tensorflow版本0.11.0rc0和另一台机器tensorflow 0.10.0rc0。对于后者，tf.batch_matrix_diag工作正常，但在前者上，我收到错误AttributeError: 'module' object has no attribute 'batch_matrix_diag' --- --- EDIT

batch_cholesky也出现了同样的错误。

有人可以解释一下如何解决这个问题吗？

Answer 1

我认为由于this change，您需要使用matrix_diag而不是batch_matrix_diag。