quantax.nn.RawInputLayer

class quantax.nn.RawInputLayer

Bases: Module

The layer that takes not only the output of the previous layer, but also the raw input basis state.

__call__(x: Array, s: Array) → Callable

The forward pass.

Parameters:

• x – The output of the previous layer.

• s – The raw input basis state.