class_docstr: Temporal Entropy Model
class_docstr_tok: ["Temporal","Entropy","Model"]
docstr: Straight-trhough round
docstr_tok: ["Straight-trhough","round"]
code: def round_st(x: Tensor) -> Tensor:
      """
      Straight-trhough round
      """
      return (torch.round(x) - x).detach() + x
  
imports: import itertools
  import sys
  from typing import Dict, List, NamedTuple, Optional, Sequence, Tuple
  import torch
  import torch.nn
  from torch import Tensor
  from .bottlenecks import GsnConditionalLocScaleShift
  from .entropy_model_layers import EncoderSection, LearnedPosition, StartSym, Transformer
  from .layers_utils import make_embedding
  from .patcher import Patcher
label_desc: Use image node_id 2 for calling the VCTEntropyModel obj's underlying member method code with example usage: obj.round_st(x) and returns: unknown