lgatr.primitives.dropout

Grade dropout.

Functions

grade_dropout(x, p[, training])

Multivector dropout that drops grades independently.

lgatr.primitives.dropout.grade_dropout(x, p, training=True)

Multivector dropout that drops grades independently.

Parameters:

• x (Tensor) – Input data of shape (..., 16).

• p (float) – Dropout probability (the same for each grade).

• training (bool) – Switches between train-time and test-time behavior.

Returns:

Inputs with dropout applied, shape (..., 16).

Return type:

outputs