track_batch# torch_brain.data.track_batch(input)[source]# Wrap an array or tensor to track the batch_index. Parameters: input (Union[Tensor, ndarray]) – An array or tensor.
