tf.linalg.LinearOperatorBlockDiag