代码之家 › 专栏 › 技术社区 › yanachen

如何展平keras中的张量?

keras-layer keras tensorflow

yanachen · 技术社区 · 7 年前

我有一个张量 words_conv_bigram_pool 形状为 (?, 1, 1, 64) ,则, ? 是批次大小。我试图将张量展平为 (?, 64) 通过 Flatten()(words_conv_bigram_pool) ,则, 但它返回的形状是 (?,?) 。

>>> Flatten()(words_conv_bigram_pool)
WARNING:tensorflow:From /home/xuemeng.cyn/anaconda2/lib/python2.7/site-packages/keras/backend/tensorflow_backend.py:1264: calling reduce_prod (from tensorflow.python.ops.math_ops) with keep_dims is deprecated and will be removed in a future version.
Instructions for updating:
keep_dims is deprecated, use keepdims instead
<tf.Tensor 'flatten_1/Reshape:0' shape=(?, ?) dtype=float32>

发生了什么以及如何在keras中使用展平功能?

1 回复 | 直到 7 年前

Daniel Möller 7 年前

不要太在意“tensorflow形状”,要在意“keras形状”。这种情况很常见 ? 使用keras时的tensorflow维度。

如果将此展平层添加到模型中,然后执行 model.summary() ,您将看到所需的形状。

如果您不使用“keras模型”,但只想删除附加尺寸,可以尝试 tf.squeeze 。

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

2 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

2 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

3 年前

Chris Ranck · Keras时间序列生成器-输入为“X”长,输出为“X-长”

3 年前

DPM · 将自定义掩码传递给LSTM数据以进行培训和验证

3 年前

Luciana Carota · 为什么Keras的R“预测”函数不能识别正确的输入维度?

3 年前

Sarah Elnaggar · 我使用Gensim Doc2vec进行图形嵌入,然后在keras中使用两层深度神经网络进行二元分类

3 年前

user836026 · 在pytorch中复制权重时语法无效

3 年前

mad · 如何从一个定制的神经网络模型中获得逻辑和概率

3 年前

Archit · 无法使用自定义距离函数加载带有lambda函数的keras模型

6 年前