tf.layers.DropoutLayer module

tf.layers.DropoutLayer module

class tf.layers.DropoutLayer.DropoutLayer(*args: Any, **kwargs: Any)

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

Wrapper around the Keras dropout layer.

call(inputs, training=True, **kwargs)

Performs the dropout.

Parameters
  • inputs (Tensor) – Arbitrary tensor.

  • training (bool) – Training mode if set to True.

Returns

A tensor of same shape as input.

Return type

Tensor