如何使用TensorFlow的WALSMatrixFactorization

Question

我正在尝试使用TensorFlow的WALSMatrixFactorization估算器，但我无法弄清楚如何使用它。 fit方法将input_fn作为参数，但该函数应该返回什么？作为输入，我基本上有一个矩阵，我想用WALS方法分解，但我不知道如何将这个矩阵传递给fit函数。