我可以使用tf.sparse_split在可变数量的零件中动态分割张量？

Question

我在Tensorflow中有一个3D稀疏张量，我想沿第一维（轴= 0）分割。我在考虑使用tf.sparse_split操作。但它需要num_splits作为Python integer。我想知道我是否在标量占位符中有num_splits有没有办法使用它？

为什么这个函数遵循了这样的约定，我还没有在任何其他tensorflow操作中看到这个？

Answer 1

在张量流框架中，必须在图形构建时知道num_splits，因为图形是静态的。至少在使用daddy的旧的基于图形的张量流时。如果你真的必须让你的图形的一部分是动态的，你可能会成功使用tensorflow的命令式eager execution。