mask_tensor

hypercoil.functional.utils.mask_tensor(tensor: Tensor, mask: Tensor, axis: Sequence[int], fill_value: float | Tensor = 0)