tf.ReshapeLayer module

tf.ReshapeLayer module

class tf.ReshapeLayer.ReshapeLayer(*args: Any, **kwargs: Any)

Bases: modelzoo.common.layers.tf.BaseLayer.BaseLayer

Wrapper around the Keras layer that reshapes the input.

call(input, **kwargs)

Apply the reshape layer to an input.

Parameters

inputs (Tensor) – A tensor.

Returns

The tensor after reshape.

Return type

Tensor