track_mask2d#

torch_brain.data.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 (Union[Tensor, ndarray]) – An array or tensor.