mirror of
https://github.com/dragen1860/TensorFlow-2.x-Tutorials.git
synced 2021-05-12 18:32:23 +03:00
27 lines
791 B
Python
27 lines
791 B
Python
import tensorflow as tf
|
|
import numpy as np
|
|
|
|
|
|
def uniform(shape, scale=0.05, name=None):
|
|
"""Uniform init."""
|
|
initial = tf.random.uniform(shape, minval=-scale, maxval=scale, dtype=tf.float32)
|
|
return tf.Variable(initial, name=name)
|
|
|
|
|
|
def glorot(shape, name=None):
|
|
"""Glorot & Bengio (AISTATS 2010) init."""
|
|
init_range = np.sqrt(6.0/(shape[0]+shape[1]))
|
|
initial = tf.random.uniform(shape, minval=-init_range, maxval=init_range, dtype=tf.float32)
|
|
return tf.Variable(initial, name=name)
|
|
|
|
|
|
def zeros(shape, name=None):
|
|
"""All zeros."""
|
|
initial = tf.zeros(shape, dtype=tf.float32)
|
|
return tf.Variable(initial, name=name)
|
|
|
|
|
|
def ones(shape, name=None):
|
|
"""All ones."""
|
|
initial = tf.ones(shape, dtype=tf.float32)
|
|
return tf.Variable(initial, name=name) |