Tensorflow:将未知大小的张量拆分为给定大小的块

时间:2019-08-30 07:38:46

标签: python tensorflow split

我想将一个张量分成相等或接近相等大小的多个子张量。由于我不知道张量的大小,因此无法始终保证将其分成偶数大小的块。但是,tf.split似乎会分裂,因此有时会失败!

在numpy中,存在np.split,如果无法将数组分为大小均匀的块,也会引发异常。为避免此问题,可以使用np.array_split,它允许最后一个不同大小的块。这正是我在张量流中寻找的行为。

总结:如果我现在不需要张量的大小,而只有所需的块大小,那么在张量流中将张量分成多个块的最佳方法是什么?我还没有找到与np.array_split类似的功能?

0 个答案:

没有答案