perf: use torch.embedding for type embedding#4747

iProzd merged 1 commit intodeepmodeling:develdeepmodeling/deepmd-kit:develfrom

caic99:perf-embeddingcaic99/deepmd-kit:perf-embeddingCopy head branch name to clipboard