fops.py 文件源码

def __init__(self, batch_size, mem_size, mem_dim, name,
                 activation=tanh, dummy_value=0.0, stddev=0.5):
        self._mem_size = mem_size
        self._mem_dim = mem_dim
        self._activation = activation
        self._batch_size = batch_size

        # memory
        M_init = tf.get_variable("%s_M_init"%name, [1, self._mem_size, self._mem_dim], tf.float32,
                            tf.random_normal_initializer(stddev=stddev))
        self._memory = tf.tile(M_init, [batch_size, 1, 1], name='%s_Tile_M'%name)