Python tf.nn.embedding_lookup用法及代码示例

参数

• params 表示完整嵌入张量的单个张量，或除第一个维度外所有相同形状的张量列表，表示遵循"div" 分区策略的分片嵌入张量。
• ids 类型为 int32 或 int64 的 Tensor 包含要在 params 中查找的 ID。
• max_norm 如果不是 None ，则每个嵌入在其 l2-norm 大于此值时被剪裁。
• name 操作的名称(可选)。

返回

• A Tensor与中的张量具有相同的类型params.

  例如，如果 params 是一个 5x2 矩阵：

  [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]

  或矩阵列表：

  params[0]:[[1, 2], [3, 4]]
  params[1]:[[5, 6], [7, 8]]
  params[2]:[[9, 10]]

  ids 是：

  [0, 3, 4]

  输出将是一个 3x2 矩阵：

  [[1, 2], [7, 8], [9, 10]]

抛出

• ValueError 如果params 为空。