keraflow.layers.core.PermuteDims Class Reference

Permutes the dimensions of the input tensor according to a given pattern. More...

Inheritance diagram for keraflow.layers.core.PermuteDims:

Public Member Functions
def	__init__

Detailed Description

Permutes the dimensions of the input tensor according to a given pattern.

Behaves like numpy.transpose().

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

Constructor & Destructor Documentation

def keraflow.layers.core.PermuteDims.__init__	(		self,
			dims,
			include_batch_dim = False,
			kwargs
	)

Parameters

dims	list of int. The permuting dimension pattern.
include_batch_dim	boolean. If true, in dims, 0 refers to the first dimension; otherwise, 0 refers to the second dimension(0->1, 1->2...).
kwargs	see Layer.__init__.