我正在使用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(...
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)