ConditionalChoice# class torch_brain.transforms.ConditionalChoice(condition, true_transform, false_transform)[source]# Bases: object Conditionally apply a single transformation based on whether a condition is met. Parameters: condition (Callable) – callable that takes a data object and returns a boolean true_transform (Callable) – transformation to apply if the condition is met false_transform (Callable) – transformation to apply if the condition is not met
