|
Keraflow
Deep Learning for Python.
|
Reshapes the input tensor according to a given pattern. More...
Public Member Functions | |
| def | __init__ |
Reshapes the input tensor according to a given pattern.
Behaves like numpy.reshape().
(nb_samples, x, y, ...)(nb_samples, a*y, ...) | def keraflow.layers.core.Reshape.__init__ | ( | self, | |
| target_shape, | |||
include_batch_dim = False, |
|||
| kwargs | |||
| ) |
| target_shape | list of int. Target shape. You could specify at most one -1, which will be automatically determined. |
| include_batch_dim | boolean. When False, in target_shape, 0 refers to the second dimension of the tensor, i.e. the dimension next to the batch size dimension. Otherwise, 0 refers to the first (batch) dimension. |
| kwargs | see Layer.__init__. |