Python tensor.copyの例

コード例 #1

ファイル: util_classes.py プロジェクト: puria-radmard/AL4ST

 def __init__(self,
              name: str,
              initialisation: torch.tensor,
              cache: bool = False):
     # might change cache to arbitrary length
     self.name = name
     self.attr = initialisation
     self.cache = cache
     if cache:
         self.prev_attr = initialisation.copy()

コード例 #2

ファイル: training.py プロジェクト: wsg1873/rtrl

 def act(self, obs: torch.tensor, r, done, info):
     return obs.copy(), {}