Create a weight variable with appropriate initialization :param name: weight name :param shape: weight shape :return: initialized weight variable
(name, shape)
| 3 | |
| 4 | # weight and bais wrappers |
| 5 | def weight_variable(name, shape): |
| 6 | """ |
| 7 | Create a weight variable with appropriate initialization |
| 8 | :param name: weight name |
| 9 | :param shape: weight shape |
| 10 | :return: initialized weight variable |
| 11 | """ |
| 12 | initer = tf.truncated_normal_initializer(stddev=0.01) |
| 13 | return tf.get_variable('W_' + name, |
| 14 | dtype=tf.float32, |
| 15 | shape=shape, |
| 16 | initializer=initer) |
| 17 | |
| 18 | |
| 19 | def bias_variable(name, shape): |