python - ValueError:新数组的总大小必须保持不变(从 keras 重塑)
问题描述
对于这个玩具模型:
from keras.layers import Input, Dense, Reshape
from keras.models import Model
# this is the size of our encoded representations
compression = 10
input_img = Input(shape=(28,28, ), name= "28x28")
encoded = Dense(int(np.floor(28*28/compression)), activation='relu',
name= "encoder_" + str(compression))(input_img)
decoded = Dense(units = 28*28, activation='sigmoid',
name = "28.28_decoder")(encoded)
reshape = Reshape(target_shape = (28,28), name = "28x28_reshape")(decoded)
autoencoder = Model(input_img, reshape)
我得到错误:
ValueError Traceback (most recent call last)
<ipython-input-27-04b835543369> in <module>
12 decoded = Dense(units = 28*28, activation='sigmoid',
13 name = "28.28_decoder")(encoded)
---> 14 reshape = Reshape(target_shape = (28,28), name = "28x28_reshape")(decoded)
15 autoencoder = Model(input_img, reshape)
16
~/.local/lib/python3.6/site-packages/keras/engine/base_layer.py in __call__(self, inputs, **kwargs)
472 if all([s is not None
473 for s in to_list(input_shape)]):
--> 474 output_shape = self.compute_output_shape(input_shape)
475 else:
476 if isinstance(input_shape, list):
~/.local/lib/python3.6/site-packages/keras/layers/core.py in compute_output_shape(self, input_shape)
396 # input shape known? then we can compute the output shape
397 return (input_shape[0],) + self._fix_unknown_dimension(
--> 398 input_shape[1:], self.target_shape)
399
400 def call(self, inputs):
~/.local/lib/python3.6/site-packages/keras/layers/core.py in _fix_unknown_dimension(self, input_shape, output_shape)
384 output_shape[unknown] = original // known
385 elif original != known:
--> 386 raise ValueError(msg)
387
388 return tuple(output_shape)
ValueError: total size of new array must be unchanged
我一直试图找出原因,但我不明白。查看帮助页面非常简单,因为前一层是执行重塑的足够形状。
解决方案
您解码的密集层有 28*28 单位,这将使其输出暗淡为 (,28,784),不会重塑为 28*28。
推荐阅读
- javascript - 自定义 styleColorBar 根据带有阴影阈值的符号左右移动
- celery - 将队列/代理内容映射到任务
- go - 如何迭代 google.protobuf.ListValue
- prolog - 我怎样才能得到这个代码在做什么?
- r - 在R中使用方括号的空间逆子集
- amazon-web-services - “免费套餐”实例...在 4 天内超过 1 GB 带宽的 85%...如何查看带宽使用情况?
- powerbi - 如何在 PowerBI 中对文本过滤器进行模糊匹配
- git - 我正在尝试解决本地功能分支中的 git 冲突,但发生了一些奇怪的事情
- .net - 异步的Kafka生产者不返回DeliveryReport但DeliveryResult
- python - Python while循环为变量返回0