【发布时间】：2017-11-10 18:05:02
【问题描述】：
让T 是一个形状为[n,f] 的张量，它代表一个批次。现在我想将T 沿axis=0 分割成m 张量。 m 的值取决于当前批次。我有另一个张量 I，形状为 [m,2]，它存储了指示切片应该出现在哪里的索引对。
我不确定如何“迭代”索引以应用 tf.slice。有什么想法吗？
这可以使用tf.scan 以某种方式实现吗？
【问题讨论】：
标签： tensorflow
【发布时间】：2017-11-10 18:05:02
【问题描述】：
我想您正在寻找split 函数。
【讨论】：
-
不，切片大小不同。这些大小在
I中指定。 -
@Christopher 为什么你认为 tf.split 只能处理相等的大小？
num_or_size_splits既可以是整数，也可以是具有拆分大小的张量。 -
是的，你是对的。我的错。现在的问题是无法推断
num的值。有什么建议吗？ -
@Christopher：您是如何解决无法推断 num 的事实的？
-
根据
tf.split()文档：“这要求 num_split 均分 value.shape[axis]。”