稀疏算子

稳定版 API

torch.ops.fbgemm.permute_2D_sparse_data(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

torch.ops.fbgemm.permute_1D_sparse_data(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

torch.ops.fbgemm.expand_into_jagged_permute(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

torch.ops.fbgemm.asynchronous_complete_cumsum(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

torch.ops.fbgemm.offsets_range(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

torch.ops.fbgemm.segment_sum_csr(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

torch.ops.fbgemm.keyed_jagged_index_select_dim1(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

torch.ops.fbgemm.block_bucketize_sparse_features(*args: ~typing.~_P, **kwargs: ~typing.~_P) → _T

其他 API