Keraflow
Deep Learning for Python.
 All Classes Namespaces Functions Pages
keraflow.layers.core.Flatten Class Reference

Flatten the input tensor into 1D. More...

Inheritance diagram for keraflow.layers.core.Flatten:

Public Member Functions

def __init__
 

Detailed Description

Flatten the input tensor into 1D.

Behaves like numpy.reshape().

  • input_shape: nD, (nb_samples, x, y, ...)
  • output_shape: 2D, (nb_samples, Prod(x,y,...))

Constructor & Destructor Documentation

def keraflow.layers.core.Flatten.__init__ (   self,
  include_batch_dim = False,
  kwargs 
)
Parameters
include_batch_dimboolean. If False, batch dimension is kept (i.e the output shape will be of two dimensions).
kwargssee Layer.__init__.