Fixing incorrect use of module relative class reference.
This commit is contained in:
parent
891b942890
commit
534c4744ca
@ -261,7 +261,7 @@ class AffineLayer(LayerWithParameters):
|
|||||||
self.input_dim, self.output_dim)
|
self.input_dim, self.output_dim)
|
||||||
|
|
||||||
|
|
||||||
class ReshapeLayer(layers.Layer):
|
class ReshapeLayer(Layer):
|
||||||
"""Layer which reshapes dimensions of inputs."""
|
"""Layer which reshapes dimensions of inputs."""
|
||||||
|
|
||||||
def __init__(self, output_shape=None):
|
def __init__(self, output_shape=None):
|
||||||
|
Loading…
Reference in New Issue
Block a user