  One step for the UniP (B(h) version). Alternatively, self.solver_p is used if is specified.  scale_model_input  < source > ( sample: FloatTensor *args **kwargs  ) → torch.FloatTensor  Parameters   sample (torch.FloatTensor) —