---
title: 功能描述
description: "将整数索引映射到一个固定大小的向量。"
url: https://www.hikunpeng.com/document/detail/zh/kunpenghpcs/hpckit/devg/KunpengHPCKit_developer_190.html
sourcePath: /source/zh/kunpenghpcs/hpckit/devg/KunpengHPCKit_developer_190.html
indexId: adb5046f200e141b51560f13f6b99eddc3fbe0a39385835b7d3738bede8c5e1f67
---
# 功能描述

#### 算子功能

将整数索引映射到一个固定大小的向量。


#### 算子公式


其中：

- x是输入的整数索引。
- W是一个二维张量，其形状为[num_embeddings, embedding_dim]，其中每一行表示一个嵌入向量。
- W[x]表示从权重矩阵W中取出索引为x的行作为嵌入向量。