class MLP(*args: Any, **kwargs: Any)

Bases: Module

Multi-Layer Perceptron

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

Forward

Parameters:

x – Tensor

Returns:

Tensor

class Net(*args: Any, **kwargs: Any)

Bases: Module

Network Layer

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

Forward

Parameters:

x – Tensor

Returns:

Tensor