track_mask#

torch_brain.data.track_mask(input)[source]#

Wrap an array or tensor to specify that its padding mask should be tracked.

Parameters:

input (Union[Tensor, ndarray]) – An array or tensor.