def test_MatrixDiagPart(self): if td._tf_version[:2] >= (0, 12): t = tf.matrix_diag_part(self.random(3, 4, 4, 5)) self.check(t)