Method pin_memory

common/batch.py:29–33

(self)

      feature_value.record_stream(stream)

  def pin_memory(self):
    args = {}
    for feature_name, feature_value in self.as_dict().items():
      args[feature_name] = feature_value.pin_memory()
    return self.__class__(**args)

  def __repr__(self) -> str:
    def obj2str(v):