稀疏運算子

穩定版 API

torch.ops.fbgemm.permute_2D_sparse_data(*args, **kwargs)

torch.ops.fbgemm.permute_1D_sparse_data(*args, **kwargs)

torch.ops.fbgemm.expand_into_jagged_permute(*args, **kwargs)

torch.ops.fbgemm.asynchronous_complete_cumsum(*args, **kwargs)

torch.ops.fbgemm.offsets_range(*args, **kwargs)

torch.ops.fbgemm.segment_sum_csr(*args, **kwargs)

torch.ops.fbgemm.keyed_jagged_index_select_dim1(*args, **kwargs)

torch.ops.fbgemm.block_bucketize_sparse_features(*args, **kwargs)

其他 API