我想将一个张量分成相等或接近相等大小的多个子张量。由于我不知道张量的大小,因此无法始终保证将其分成偶数大小的块。但是,tf.split
似乎会分裂,因此有时会失败!
在numpy中,存在np.split
,如果无法将数组分为大小均匀的块,也会引发异常。为避免此问题,可以使用np.array_split
,它允许最后一个不同大小的块。这正是我在张量流中寻找的行为。
总结:如果我现在不需要张量的大小,而只有所需的块大小,那么在张量流中将张量分成多个块的最佳方法是什么?我还没有找到与np.array_split
类似的功能?