for all models (such as downloading or saving).  forward  < source > ( sample: FloatTensor return_dict: bool = True  ) → VQEncoderOutput or tuple  Parameters   sample (torch.FloatTensor) — Input sample.   return_dict (bool, optional, defaults to True) —