import tensorflow as tf

from .. import var_init



def embeddings(*, id_space_size, embedding_size):
  return tf.Variable(var_init.normal([id_space_size, embedding_size]))

import tensorflow as tf

from ..variable import variable



def embeddings(*, id_space_size, embedding_size):
  return variable([id_space_size, embedding_size])

Use variable module instead of var_init
Use variable module instead of var_init
