keraflow.layers.core.ExpandDims Class Reference

Expand dimension of the input tensor. More...

Inheritance diagram for keraflow.layers.core.ExpandDims:

Public Member Functions
def	__init__

Detailed Description

Expand dimension of the input tensor.

Behaves like numpy.expand_dims().

• input_shape: nD, (nb_samples, x, | y, ...)
• output_shape: (n+1)D,(nb_samples, x, 1, y, ...)`

Constructor & Destructor Documentation

def keraflow.layers.core.ExpandDims.__init__	(		self,
			axis,
			include_batch_dim = False,
			kwargs
	)

Parameters

axis	int. The axis to expand dimension.
include_batch_dim	boolean. If true, axis=0 refers to the first dimension; otherwise, axis=0 refers to the second dimension(0->1, 1->2...).
kwargs	see Layer.__init__.