Other Functional Interfaces

RecIS also provides other functional operation interfaces:

array_ops

recis.nn.functional.array_ops.bucketize(input, boundaries)

Data bucketing

recis.nn.functional.array_ops.bucketize_mod(input, bucket)

Data modulo

recis.nn.functional.array_ops.multi_hash(input, muls, primes, bucket_num)

Multi-hash

embedding_ops

recis.nn.functional.embedding_ops.ids_partition(ids, max_partition_num, world_size)

ID partitioning function for distributed scenarios.

recis.nn.functional.embedding_ops.ids_encode(ids_list, table_ids)

ID encoding function.

recis.nn.functional.embedding_ops.merge_offsets(offsets)

Offset merging function.

recis.nn.functional.embedding_ops.ragged_embedding_segment_reduce(emb, offsets, combiner='sum')

Ragged Tensor format embedding aggregation.

recis.nn.functional.embedding_ops.sparse_embedding_segment_reduce(unique_emb, weight, reverse_indices, segment_ids, num_segments, combiner)

Sparse Tensor format embedding segment aggregation.

fused_ops

recis.nn.functional.fused_ops.fused_bucketize_gpu(values: List[torch.Tensor], boundaries: List[torch.Tensor])

Multi-group data bucketing operation processing

recis.nn.functional.fused_ops.fused_uint64_mod_gpu(values: List[torch.Tensor], mods: List | torch.Tensor)

Multi-group data modulo operation processing

recis.nn.functional.fused_ops.fused_ids_encode_gpu(ids_list: List[torch.Tensor], table_ids: torch.Tensor | list)

Multi-group data ID encoding operation processing

recis.nn.functional.fused_ops.fused_multi_hash(inputs: List[torch.Tensor], muls: List[torch.Tensor], primes: List[torch.Tensor], bucket_nums: List[torch.Tensor])

Multi-group data multi-hash operation processing

recis.nn.functional.fused_ops.fused_multi_hash(inputs: List[torch.Tensor], muls: List[torch.Tensor], primes: List[torch.Tensor], bucket_nums: List[torch.Tensor])

Multi-group data multi-hash operation processing

recis.nn.functional.ragged_ops.fused_ragged_cutoff_2D(values: List[torch.Tensor], offsets: List[torch.Tensor], keep_lengths: torch.Tensor, drop_sides: torch.Tensor, pad_sides: torch.Tensor)

Multi-group 2D ragged tensor cutoff operation processing

recis.nn.functional.ragged_ops.fused_ragged_cutoff_3D(values: List[torch.Tensor], offsets: List[torch.Tensor], keep_lengths: torch.Tensor, drop_sides: torch.Tensor, pad_sides: torch.Tensor)

Multi-group 3D ragged tensor cutoff operation processing

hash_ops

recis.nn.functional.hash_ops.farmhash(inputs, splits)

Multi-group data farm hash operation processing

recis.nn.functional.hash_ops.murmurhash(inputs, splits)

Multi-group data murmur hash operation processing