Method apply

Module/Map/Graph.py:50–52

(self, func: T.Callable[[torch.Tensor,], torch.Tensor])

            self.data[k].__setitem__(index, value.data[k])

    def apply(self, func: T.Callable[[torch.Tensor,], torch.Tensor]) -> None:
        for k in self.data:
            self.data[k] = func(self.data[k])
  
    def __len__(self) -> int:
        return self.index.size(0)