Jagged Tensor 運算子

穩定版 API

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

其他 API

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

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

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

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

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

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

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

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

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

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