Python Layer.build Examples

Example #1

def plot_custom_layer(layer: Layer, input_shape: tuple, plot_filepath: str):
    tmp_input = Input(shape=input_shape)
    layer.build(input_shape)
    tmp_output = layer.call(tmp_input)
    tmp_model = Model(inputs=[tmp_input], outputs=[tmp_output])
    return plot_model(tmp_model, to_file=plot_filepath, show_shapes=True)

Example #2

File: keras_frontend.py Project: Gilgamesh-qx/FrScatNet

 def build(self, input_shape):
     self.shape = input_shape
     Layer.build(self, input_shape)