Keras Tensorflow 中的切片张量

Question

例如，我有形状为(None, 2, 100, 100, 1024) 的张量，我想将2 拆分为1 和1，这样我就有2 个等级为4 的张量(None, 100, 100, 1024)。如何使用 Keras Tensorflow 做到这一点？

谢谢。

Answer 1

使用tf.split():

import tensorflow as tf

tensor = tf.placeholder(tf.float32, (None, 2, 100, 100, 1024))
splitted = [tf.squeeze(t, axis=1) for t in tf.split(tensor, 2, axis=1)]
print(splitted[0].get_shape().as_list(), splitted[1].get_shape().as_list())
# [None, 100, 100, 1024] [None, 100, 100, 1024]

要连接回来：

# manipulate here ...
splitted = [t[:, None, ...] for t in splitted]
res = tf.concat(splitted, axis=1)
print(res.get_shape().as_list()) # [None, 2, 100, 100, 1024]