pytagi.nn.layer_block

Classes

LayerBlock

A stack of different layers derived from BaseLayer

Module Contents

class pytagi.nn.layer_block.LayerBlock(*layers: pytagi.nn.base_layer.BaseLayer)

Bases: pytagi.nn.base_layer.BaseLayer

A stack of different layers derived from BaseLayer

Initialize the Sequential model with the given layers. :param layers: A variable number of layers (instances of BaseLayer or derived classes).

switch_to_cuda()

Convert all layers to cuda layer

property layers: None

Get layers