Method init

Module/Map/Graph.py:33–38

(cls: type[Self], data: dict[T_Fields, torch.Tensor])

    
    @classmethod
    def init(cls: type[Self], data: dict[T_Fields, torch.Tensor]) -> Self:
        sizes = [v.size(0) for v in data.values()]
        assert all([s == sizes[0] for s in sizes]), f"TensorBundle requires all features have (Nx...) shape with same 'N', get {sizes}"
        
        index   = torch.arange(0, sizes[0], dtype=torch.long)
        return cls(index, data)

    def __getitem__(self, index) -> TensorBundle[T_Fields]:
        selected_dict: dict[T_Fields, torch.Tensor] = {