track_mask2d# torch_brain.batching.track_mask2d(input)[source]# Wrap an array or tensor to specify that its padding mask should be tracked. This is used in conjunction with pad2d. Parameters: input (Tensor | ndarray) – An array or tensor.
