      def __call__(self, inputs: Optional[Tensor] = None) -> Tensor:
          """ Decoder Network.
  
          Parameters
          inputs: Tensor, optional
              If the input is an output from another model (such as the Fully Connected Model) this
              should be ``None`` otherwise it should be a Tensor
  
          Returns
          -------
          :class:`keras.models.Model`
              The Decoder model
 
</s> remove     def __call__(self, inputs: Optional[Tensor] = None) -> Tensor:
        """ Decoder Network.
 </s> add     def __call__(self, inputs: Union[Tensor, List[Tensor]]) -> Union[Tensor, List[Tensor]]:
        """ Upscale Network. </s> remove         :class:`keras.models.Model`
            The Decoder model
 </s> add          Tensor or list of tensors
            The output of encoder blocks. Either a single tensor (if learn mask is not enabled) or
            list of tensors (if learn mask is enabled) </s> remove     input_shape: tuple
        The shape tuple for the input to the decoder.
 </s> add  </s> remove     def __init__(self, input_shape: Tuple[int, int, int], config: dict) -> None:
 </s> add     def __init__(self, input_shape: Tuple[int, ...], config: dict) -> None:
        inputs: Tensor or list of tensors
            Input tensor(s) to upscale block. This will be a single tensor if learn mask is not
            selected or if this is the first call to the upscale blocks. If learn mask is selected
            and this is not the first call to upscale blocks, then this will be a list of the face
            and mask tensors.
      def __call__(self, inputs: Optional[Tensor] = None) -> Tensor:
          """ Decoder Network.
  
          Parameters
        inputs: Tensor or list of tensors
            Input tensor(s) to upscale block. This will be a single tensor if learn mask is not
            selected or if this is the first call to the upscale blocks. If learn mask is selected
            and this is not the first call to upscale blocks, then this will be a list of the face
            and mask tensors.
        inputs: Tensor or list of tensors
            Input tensor(s) to upscale block. This will be a single tensor if learn mask is not
            selected or if this is the first call to the upscale blocks. If learn mask is selected
            and this is not the first call to upscale blocks, then this will be a list of the face
            and mask tensors.
        inputs: Tensor or list of tensors
            Input tensor(s) to upscale block. This will be a single tensor if learn mask is not
            selected or if this is the first call to the upscale blocks. If learn mask is selected
            and this is not the first call to upscale blocks, then this will be a list of the face
            and mask tensors.
  
          Returns
          -------
          :class:`keras.models.Model`
              The Decoder model
keep keep keep keep replace replace replace keep keep keep keep keep
 Bugfix: Phaze-A Fix learn mask for upscales in FC 
plugins/train/model/phaze_a.py