model.tools.layers.Store

Contents

model.tools.layers.Store#

class Store(storage: Storage)[source]#

Bases: LayerGen

Stores a tensor in the specified storage and passes it on

Methods

get

Initializes and returns the network layer

get(in_channels: int) Tuple[Module, int][source]#

Initializes and returns the network layer

Parameters:

in_channels (int) – Number of input channels.

Returns:

The generated module and the number of channels that will be after applying this layer to a tensor with in_channels channels.

Return type:

Tuple[nn.Module, int]