LinearAEDecoder

class behavenet.models.aes.LinearAEDecoder(n_latents, output_size, encoder=None)[source]

Bases: BaseModule

Linear decoder.

Methods Summary

build_model()

Construct the decoder.

forward(x[, dataset])

Process input data.

Methods Documentation

build_model()[source]

Construct the decoder.

forward(x, dataset=None)[source]

Process input data.

Parameters:

  • x (torch.Tensor object) – input data

  • dataset (int) – used with session-specific io layers

Returns:

shape (n_input_channels, y_pix, x_pix)

Return type:

torch.Tensor