class_docstr: Abstract base class for UCB-style CB models.
class_docstr_tok: ["Abstract","base","class","for","UCB-style","CB","models","."]
code: def input_prototype(self) -> torch.Tensor:
      return torch.randn(1, self.input_dim)
  
file_path: ReAgent/reagent/models/cb_base_model.py
imports: import logging
  from abc import ABC, abstractmethod
  from typing import Dict, Optional
  import torch
label_desc: Use image node_id 2 for calling the UCBBaseModel obj's underlying member method code with example usage: obj.input_prototype() and returns: torch