Pooled Embedding 运算符

稳定 API

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

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

其他 API