我想将张量分成大小相等或接近相等的多个子张量。由于我事先不知道张量的大小，因此并不总是保证分裂成大小均匀的块。然而，tf.split似乎期望均匀分裂，因此有时会失败！
在 numpy 中，np.split如果数组不能分成大小均匀的块，它也会引发异常。为了避免这个问题，可以使用np.array_split，它允许使用不同大小的最后一个块。这正是我在 tensorflow 中寻找的行为。
总结：如果我现在不知道张量的大小而只有所需的块大小，那么在 tensorflow 中将张量分成多个块的最佳方法是什么？是否有类似的功能np.array_split我还没有找到？