Source code for mrnntorch.region.input_region

from mrnntorch.region.region_base import Region, DEFAULT_REGION_BASE




class InputRegion(Region):
    def __init__(
        self,
        num_units,
        sign=DEFAULT_REGION_BASE["sign"],
        device=DEFAULT_REGION_BASE["device"],
    ):
        """Initialize an input region.

        Args:
            num_units (int): Number of input channels.
            sign (str): "pos" or "neg" indicating sign mask for inputs.
            device (str): Torch device string.
        """
        # Implements base region class
        super(InputRegion, self).__init__(num_units, sign=sign, device=device)