def variable(value, dtype=None, borrow=None, broadcastable=None, name=None, settable=True):
if dtype is None:
dtype = np.float32
x = tf.Variable(value, dtype=tf.as_dtype(dtype), name=name)
x._statestream_settable = settable
if settable:
x._assign_placeholder = tf.placeholder(dtype, shape=value.shape)
x._assign_op = x.assign(x._assign_placeholder)
return x
评论列表
文章目录