ScaleLinear

class lightning_ir.models.cross_encoders.mono.ScaleLinear(in_features: int, out_features: int, bias: bool = True, device=None, dtype=None)

Bases: Linear

Methods

forward(input)

Runs the forward pass.

Attributes

in_features
out_features
weight
training

forward(input: Tensor) → Tensor

Runs the forward pass.