def ones(shape, dtype=None, name=None):
if dtype is None:
dtype = np.float32
return tf.ones(shape=shape, dtype=tf.as_dtype(dtype), name=name)
微信
公众号
扫码关注公众号