conform_mask

hypercoil.functional.utils.conform_mask(tensor: Tensor, mask: Tensor, axis: Sequence[int], batch=False) → Tensor

Conform a mask or weight for elementwise applying to a tensor.

There is almost certainly a better way to do this.

See also

apply_mask()