sketchgraphs_models.torch_extensions.segment_ops

Classes

SegmentArgmaxPython
SegmentLogsumexpPython
SegmentLogsumexpScatter

Functions

sketchgraphs_models.torch_extensions.segment_ops.segment_argmax(values, scopes, sparse_grad=True)

sketchgraphs_models.torch_extensions.segment_ops.segment_argmax_backward(grad_output, argmax, scopes, input_shape, sparse_grad=True)

sketchgraphs_models.torch_extensions.segment_ops.segment_argmax_loop(values, scopes)

sketchgraphs_models.torch_extensions.segment_ops.segment_argmax_python(values, scopes, sparse_grad=True)

sketchgraphs_models.torch_extensions.segment_ops.segment_argmax_scatter(values, scopes, sparse_grad=True)

sketchgraphs_models.torch_extensions.segment_ops.segment_logsumexp(values, scopes)

sketchgraphs_models.torch_extensions.segment_ops.segment_logsumexp_backward_python(grad_output, values, logsumexp, lengths)

sketchgraphs_models.torch_extensions.segment_ops.segment_logsumexp_python(values: torch.Tensor, scopes: torch.Tensor)

sketchgraphs_models.torch_extensions.segment_ops.segment_logsumexp_scatter(values, scopes)

sketchgraphs_models.torch_extensions.segment_ops.segment_op_python(values, scopes, op)