diff --git a/mlp/layers.py b/mlp/layers.py index 7bb1966..b7095d8 100644 --- a/mlp/layers.py +++ b/mlp/layers.py @@ -261,7 +261,7 @@ class AffineLayer(LayerWithParameters): self.input_dim, self.output_dim) -class ReshapeLayer(layers.Layer): +class ReshapeLayer(Layer): """Layer which reshapes dimensions of inputs.""" def __init__(self, output_shape=None):