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 為空。