Feature๏
- class dgl.graphbolt.Feature[source]๏
Bases:
objectA wrapper of feature data for access.
- metadata()[source]๏
Get the metadata of the feature.
- Returns:
The metadata of the feature.
- Return type:
Dict
- read(ids: Tensor | None = None)[source]๏
Read from the feature.
- Parameters:
ids (torch.Tensor, optional) โ The index of the feature. If specified, only the specified indices of the feature are read. If None, the entire feature is returned.
- Returns:
The read feature.
- Return type:
torch.Tensor
- size()[source]๏
Get the size of the feature.
- Returns:
The size of the feature.
- Return type:
torch.Size
- update(value: Tensor, ids: Tensor | None = None)[source]๏
Update the feature.
- Parameters:
value (torch.Tensor) โ The updated value of the feature.
ids (torch.Tensor, optional) โ The indices of the feature to update. If specified, only the specified indices of the feature will be updated. For the feature, the ids[i] row is updated to value[i]. So the indices and value must have the same length. If None, the entire feature will be updated.