sketchgraphs_models.torch_extensions.index

Functions

sketchgraphs_models.torch_extensions.index.segment_cartesian_product(values_a, values_b, scopes_a, scopes_b)

sketchgraphs_models.torch_extensions.index.segment_cartesian_product_loop(values_a, values_b, scopes_a, scopes_b)

sketchgraphs_models.torch_extensions.index.segment_triu_indices(scopes, offset=0, dtype=torch.int64, device='cpu')

sketchgraphs_models.torch_extensions.index.segment_triu_indices_loop(scopes, offset=0, dtype=torch.int64, device='cpu')