code: def model_type(dtype):
      "Return the torch type corresponding to `dtype`."
      return (torch.float32 if np.issubdtype(dtype, np.floating) else
              torch.int64 if np.issubdtype(dtype, np.integer)
              else Nonef model_type(dtype):
      "Return the torch type corresponding to `dtype`."
      return (torch.float32 if np.issubdtype(dtype, np.floating) else
              torch.int64 if np.issubdtype(dtype, np.integer)
              else None)
docs: Return the torch type corresponding to `dtype`.