ApplyMask

class hybrid_learning.datasets.transforms.tuple_transforms.ApplyMask[source]

Bases: Transform

Given a 3-tuple of input, target, mask apply the mask to the first two. Returns a two-tuple of flat tensors.

Public Data Attributes:

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, mask)

Application of transformation.

Inherited from : py: class:Transform

apply_to(prediction, target, mask)

Application of transformation.

Special Methods:

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.


apply_to(prediction, target, mask)[source]

Application of transformation.

Parameters
Return type

Tuple[Tensor, Tensor]