boundlab.diff.zono3.diff_heaviside_pruning_handler#

boundlab.diff.zono3.diff_heaviside_pruning_handler(scores, data)[source]#

Differential handler for boundlab::heaviside_pruning.

Parameters:

  • scores – DiffExpr3 / DiffExpr2 providing pruning scores. Only the y component participates in masking.

  • data – DiffExpr3 / DiffExpr2 providing the tensor to prune.