RAMjET

ramjet.models.nyx

Code for a model with gaps in batch normalization usage.

Module Contents

Classes

Nyx A model with gaps in batch normalization usage.
class Nyx[source]

Bases: tensorflow.keras.Model

A model with gaps in batch normalization usage.

__init__(self)[source]

Initialize self. See help(type(self)) for accurate signature.

call(self, inputs, training=False, mask=None)[source]

The forward pass of the layer.

Parameters:
  • inputs – The input tensor.
  • training – A boolean specifying if the layer should be in training mode.
  • mask – A mask for the input tensor.
Returns:

The output tensor of the layer.