Function XInputEmbeddingLookup2

src/model/encoder/embedding_lookup_encoder.cc:37–45

randn initialization (0, 1e-3) for graph-ctr models

// randn initialization (0, 1e-3) for graph-ctr models
GraphNode* XInputEmbeddingLookup2(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_RANDN, 0, 1e-3);
  return EmbeddingLookup("", X, W);
}

// rand initiialization (-1 / col, 1 / col)
GraphNode* XInputEmbeddingLookup3(const std::string& prefix, GraphNode* X,