def scalar(value, dtype=np.float32, borrow=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=x.get_shape().as_list())
x._assign_op = x.assign(x._assign_placeholder)
return x
评论列表
文章目录