Advertisement
Guest User

Untitled

a guest
Jun 25th, 2019
71
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.70 KB | None | 0 0
  1. import tensorflow as tf
  2. from tensorflow.python.framework.ops import enable_eager_execution
  3. enable_eager_execution()
  4.  
  5.  
  6.  
  7. class MyDenseLayer(tf.keras.layers.Layer):
  8. def __init__(self, num_outputs):
  9. super(MyDenseLayer, self).__init__()
  10. self.num_outputs = num_outputs
  11.  
  12. def build(self, input_shape):
  13. X = tf.random.uniform([int(input_shape[-1]),self.num_outputs],minval=0,maxval=1,dtype=tf.dtypes.float32,)
  14. k = tf.Variable(X, name="kernel")
  15. self.kernel = 0.5 * (k+tf.transpose(k))
  16.  
  17. def call(self, input):
  18. return tf.matmul(input, self.kernel)
  19.  
  20.  
  21. layer = MyDenseLayer(5)
  22. print(layer(tf.ones([3, 5])))
  23. print(layer.trainable_variables)
  24.  
  25. print(layer.trainable_variables)
  26.  
  27. []
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement