lgatr.layers.dropout.GradeDropout

class lgatr.layers.dropout.GradeDropout(p=0.0)[source]

Bases: Module

Dropout on multivectors.

Parameters:

p (float) – Dropout probability.

forward(multivectors, scalars)[source]

Forward pass. Applies dropout.

Parameters:

  • multivectors (torch.Tensor) – Multivector inputs with shape (…, 16).

  • scalars (torch.Tensor) – Scalar inputs with shape (…).

Return type:

Tuple[Tensor, Tensor]

Returns:

  • outputs_mv (torch.Tensor) – Multivector inputs with dropout applied, shape (…, 16).

  • output_scalars (torch.Tensor) – Scalar inputs with dropout applied, shape (…).