import tensorflow as tf

        init_sd = 1.0 / np.sqrt(self.embedding_size)
        # Embedding matrices for entities and relationship types
        head_init = tf.truncated_normal([head_cnt, self.embedding_size], stddev=init_sd)
        rel_init = tf.truncated_normal([rel_cnt, self.embedding_size], stddev=init_sd)
