You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
11 lines
446 B
11 lines
446 B
import tensorflow as tf |
|
import numpy as np |
|
|
|
def weight_variable_glorot(input_dim, output_dim, name=""): |
|
"""Create a weight variable with Glorot & Bengio (AISTATS 2010) |
|
initialization. |
|
""" |
|
init_range = np.sqrt(6.0 / (input_dim + output_dim)) |
|
initial = tf.random_uniform([input_dim, output_dim], minval=-init_range, |
|
maxval=init_range, dtype=tf.float32) |
|
return tf.Variable(initial, name=name)
|
|
|