track_mask# torch_brain.batching.track_mask(input)[source]# Wrap an array or tensor to specify that its padding mask should be tracked. Parameters: input (Tensor | ndarray) – An array or tensor.
