请问关于input_shape =??的问题

来源:2-17 实战sklearn封装keras模型

慕工程6078732

2020-05-01

请问关于input_shape =??的问题

= 右边的格式要求是什么呢?似乎一定要是一个 iterable的变量对吗?比如tuple?
我尝试 input_shape = x_train.shape[1] 报错 error:‘int’ object is not iterable?
我明白 x_train.shape[1:] return (8,)

尝试看到tensorflow.org里面的其他例子有的写 input_shape = (None, 8), 感觉是在对应training graph的dimention, 可否解释一下如何对应对应的维度?谢谢

写回答

1回答

正十七

2020-05-13

  1. 格式的话,应该是tuple或者list都行,你可以试试list

  2. 如果是两个维度的话,那么第一个维度是batch_size, 第二个维度就是数据的维度。如果是四个维度,那么输入就是多张图像,第一个维度同样是batch_size, 后三个维度代表图像。

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程