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

问题描述 投票:0回答:1

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

我制作的模型是在tf.keras.Conv2D的序列中(这需要4D输入张量(样本,行,列,通道)] >>

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

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

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

我正在使用tensorflow ver 2,tensorflow.keras。我制作的模型按序列tf.keras.Conv2D(需要4D输入张量(样本,行,列,通道,通道),然后是tf.keras.convLSTM2D(...

tensorflow keras tensorflow2.0
1个回答
0
投票
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)
© www.soinside.com 2019 - 2024. All rights reserved.