frarch.models.mock module

class frarch.models.mock.BypassModel

Bases: torch.nn.modules.module.Module

Bypass mock model.

Module that returns the same argument that is given.

forward(x: torch.Tensor) → torch.Tensor

Compute forward for bypass model.

Parameters

x (torch.Tensor) – input to the model.

Returns

same tensor as the input.

Return type

torch.Tensor

training: bool