def build(self, input_shape):
时间: 2023-12-03 11:07:19 浏览: 119
自定义shape
3星 · 编辑精心推荐
`build(self, input_shape)` 是 Keras 模型中的一个方法,用于在模型第一次被调用前创建模型的权重。
参数 `input_shape` 是输入张量的形状(不包括批量大小)。该方法可以根据输入形状来创建模型所需的权重和变量。
在 `build` 方法中,通常会定义模型的各层和各层的权重。这些权重将会在后续的训练中被更新。
阅读全文