Function XOutputEmbeddingLookup

src/model/encoder/embedding_lookup_encoder.cc:62–70

/ output embedding lookup functions */ /

/* output embedding lookup functions */
/************************************************************************/
GraphNode* XOutputEmbeddingLookup(const std::string& prefix, GraphNode* X,
                                  const GroupConfigItem3& item, int sparse) {
  DXCHECK_THROW(X->shape().is_rank(2));
  int tensor_type = sparse ? TENSOR_TYPE_SRM : TENSOR_TYPE_TSR;
  auto* W =
      GetVariable(prefix + "W", Shape(item.embedding_row, item.embedding_col),
                  tensor_type, TENSOR_INITIALIZER_TYPE_ZEROS, 0, 0);
  return EmbeddingLookup("", X, W);
}

}  // namespace embedx