Jagged Tensor Operators

Stable API

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

Other API

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

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

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

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

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

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

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

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

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

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