sakura.utils.gradient_reverse.NeutralizeLayerF

class sakura.utils.gradient_reverse.NeutralizeLayerF(*args, **kwargs)

Bases: Function

Custom autograd Function that nullifies gradients during backward pass

During forward:

• Acts as identity operation (returns input unchanged)

During backward:

• Returns zero gradients, stopping gradient flow

Methods

backward	Backward pass that nullifies upstream gradients.
forward	Forward pass for gradient neutralization layer (identity transform).

Attributes