UnpackMask

class hybrid_learning.datasets.transforms.tuple_transforms.UnpackMask(selected_index=0)

Bases: _SingleIndexTupleTrafo

Undo a PackMask operation on the target. Accepts a two-tuple of tensors and returns a three-tuple of (pred, target, bool_mask).

Public Data Attributes:

Inherited from : py: class:_SingleIndexTupleTrafo

settings

Settings to reproduce the instance.

Inherited from : py: class:Transform

IDENTITY_CLASS	The identity class or classes for composition / addition.
settings	Settings to reproduce the instance.

Public Methods:

apply_to(prediction, target_and_mask)

Application of transformation.

Inherited from : py: class:Transform

apply_to(prediction, target_and_mask)

Application of transformation.

Special Methods:

Inherited from : py: class:_SingleIndexTupleTrafo

__init__([selected_index])

Init.

Inherited from : py: class:Transform

__repr__()	Return repr(self).
__eq__(other)	Return self==value.
__copy__()	Return a shallow copy of self using settings.
__add__(other)	Return a flat composition of self with other.
__radd__(other)	Return a flat composition of other and self.
__call__(*inps)	General call to the transformation.

---

Parameters

selected_index (int) –

apply_to(prediction, target_and_mask)

Application of transformation.

Parameters

• prediction (Tensor) –

• target_and_mask (Tensor) –

Return type

Tuple[Tensor, Tensor, BoolTensor]