MaskNestedTensors

class torchaug.ta_tensors.nested.MaskNestedTensors(tensors)[source]

Implement Masks Nested Tensor for PyTorch.

batch_tensors_type

alias of BatchMasks

tensors_type

alias of Mask

to_batch()[source]

Return the batched mask of the nested masks.

Return type:

BatchMasks