torch_geometric.transforms.IndexToMask

class IndexToMask(attrs: Optional[Union[str, List[str]]] = None, sizes: Optional[Union[int, List[int]]] = None, replace: bool = False)

Bases: BaseTransform

将索引转换为掩码表示 （功能名称：index_to_mask）。

Parameters:

• attrs (str, [str], optional) – 如果给定，将仅对给定的属性执行索引到掩码的转换。如果省略，将从后缀 _index 推断属性。（默认值：None）

• sizes (int, [int], optional) – 掩码的大小。如果设置为 None，将返回一个自动调整大小的张量。默认情况下将使用节点的数量， 但对于边属性，将使用边的数量作为掩码大小。 (默认值: None)

• replace (bool, 可选) – 如果设置为 True，则用掩码张量替换索引属性。（默认值：False）