track_mask2d8#

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

Wrap an array or tensor to specify that its padding mask should be tracked. This is used in conjunction with pad2d8.

Parameters:

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