Fixing incorrect use of module relative class reference.

This commit is contained in:
Matt Graham 2016-11-02 03:36:53 +00:00
parent 891b942890
commit 534c4744ca

View File

@ -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):