温馨提示:本文翻译自stackoverflow.com,查看原文请点击:tensorflow - How to input 5D tensor to keras model.fit
keras tensorflow tensorflow2.0

tensorflow - 如何将5D张量输入到keras model.fit

发布于 2020-03-27 16:17:38

我正在使用tensorflow ver 2,tensorflow.keras。

我制作的模型按顺序排列tf.keras.Conv2D(需要4D输入张量(样本,行,列,通道)

然后tf.keras.convLSTM2D(需要5D输入张量(样本,时间,行,列,通道)。

由于这个原因,我使用5D张量(样本,时间,行,列,通道,通道)进行了输入,但是在实现时一开始就无法将其输入tf.keras.Conv2D中。 model.fit(train_data, train_data... )

有什么方法可以使model.fit采取5D张量?

查看更多

查看更多

提问者
Brian Lee
被浏览
206
Orphee Faucoz 2020-01-31 17:20

您需要TimeDistributed像中实现conv2D:

x_conv = tf.keras.layers.TimeDistributed(tf.keras.layers.Conv2D(filters=filters,
                                                                kernel_size=kernel_size,
                                                                strides=strides,
                                                                padding='same',
                                                                kernel_initializer='he_normal'))(x)

通过这种方式,图层可以了解到您正在提供4D输入 timestep