在 TF 2.0 Beta 中,我正在尝试:
x = tf.keras.layers.Input(shape=(240, 2), dtype=tf.float32)
print(x.shape) # (None, 240, 2)
a = x[:, :, 0]
print(a.shape) # <unknown>
在 TF 1.x 中,我可以这样做:
x = tf1.placeholder(tf1.float32, (None, 240, 2)
a = x[:, :, 0]
它会正常工作。如何在 TF 2.0 中实现这一点?我认为
tf.split(x, 2, axis=2)
可能有效,但是我想使用切片而不是硬编码 2(轴 2 的暗淡)。
慕工程0101907
相关分类