torch_concepts.nn.functional.selection_eval

selection_eval(selection_weights: Tensor, *predictions: Tensor) Tensor[source]

Evaluate concept selection by computing weighted predictions.

Parameters:
  • selection_weights – Weights for selecting between predictions.

  • *predictions – Variable number of prediction tensors to combine.

Returns:

Weighted combination of predictions.

Return type:

Tensor