class_docstr: A type alias for a masking functions.
  
  Forward-propagate a mask in a layer of a finite-width network.
class_docstr_tok: ["A","type","alias","for","a","masking","functions",".","Forward-propagate","a","mask","in","a","layer","of","a","finite-width","network","."]
code: def __call__(
      self,
      mask: Union[jnp.ndarray, Sequence[jnp.ndarray]],
      input_shape: Shapes,
  ) -> Union[jnp.ndarray, Sequence[jnp.ndarray]]:
      ...
  
imports: from typing import Any, Generator, Optional, Protocol, Sequence, TYPE_CHECKING, TypeVar, Union
  import jax
  import jax.numpy
  from .kernel import Kernel
label_desc: Use image node_id 1 for calling the MaskFn obj's underlying member method code with example usage: obj.__call__(mask, input_shape) without return types