池化嵌入操作符

稳定版 API

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

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

其他 API