Advertisement
Guest User

Untitled

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