Fix: nn.Parameter return type identified as Tensor instead of nn.Parameter

Fix: nn.Parameter return type identified as Tensor instead of nn.Parameter #378251