torch_geometric.transforms.MaskToIndex

class MaskToIndex(attrs: Optional[Union[str, List[str]]] = None, replace: bool = False)[source]

Bases: BaseTransform

将掩码转换为索引表示 （函数名称：mask_to_index）。

Parameters:

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

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