mirror of
https://github.com/zebrajr/tensorflow.git
synced 2026-01-15 12:15:41 +00:00
added Returns and Raises for Reshape class
This commit is contained in:
committed by
GitHub
parent
a3ad605e87
commit
52e288cafa
@@ -492,6 +492,12 @@ class Reshape(Layer):
|
||||
target_shape: Target shape. Tuple of integers, does not include the
|
||||
samples dimension (batch size).
|
||||
**kwargs: Any additional layer keyword arguments.
|
||||
|
||||
Returns:
|
||||
A layer to Reshape the input.
|
||||
|
||||
Raises:
|
||||
ValueError: If input_shape and target_shape are of different shape.
|
||||
"""
|
||||
super(Reshape, self).__init__(**kwargs)
|
||||
self.target_shape = tuple(target_shape)
|
||||
|
||||
Reference in New Issue
Block a user