使用tf.keras.MaxPooling1D出现错误

最新推荐文章于 2023-12-12 18:45:58 发布

IMWTJ

最新推荐文章于 2023-12-12 18:45:58 发布

阅读量1.7k

点赞数

分类专栏： tensorflow python学习文章标签： tensorflow

本文链接：https://blog.csdn.net/imwtj123/article/details/111505651

版权

在使用tf.keras.MaxPooling1D时遇到ValueError：'Negative dimension size caused by subtracting 2 from 1 for 'pool_2/MaxPool'。错误源于通道(channels_last vs channels_first)数据格式问题。解决方案包括修改keras.json配置文件或在MaxPooling1D中明确指定data_format='channels_first'。

摘要由CSDN通过智能技术生成

错误如下：

ValueError: Negative dimension size caused by subtracting 2 from 1 for 'pool_2/MaxPool' (op: 'MaxPool') with input shapes: [?,1,1,32].

首先了解MaxPooling1D：

tf.layers.max_pooling1d(
    inputs,
    pool_size,
    strides,
    padding='valid',
    data_format='channels_last',
    name=None
)

用于1维输入的MaxPooling层

pool_size:表示pooling window的大小
strides:指定pooling操作的步长
padding:一个字符串。padding的方法：string，valid或same，大小写不敏感。
data_format:一个字符串，channels_last（默认）或channels_first中的一个，输入中维度的排序，channels_last对应于具有形状(batch, length, channels)的输入，而channels_first对应于具有形状(batch, channels, length)的输入。
name:一个字符串，表示层的名称。