标签: python tensorflow
我有一个张量列表,每个元素都有不同的形状。例如,我的列表中有两个张量,第一个张量的形状为3 3 3,第二个张量的形状为4 * 4。我想在TensorFlow中从它们中随机抽取一个张量。但是我不知道该怎么做。
我现在的方法是将所有张量整形为1 * N并使用tf.concat创建一个新的张量。然后,我可以使用tf.gather。但这太慢了。我想直接使用索引来选择张量。