Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import tensorflow as tf
- from tensorflow.python.framework.ops import enable_eager_execution
- enable_eager_execution()
- class MyDenseLayer(tf.keras.layers.Layer):
- def __init__(self, num_outputs):
- super(MyDenseLayer, self).__init__()
- self.num_outputs = num_outputs
- def build(self, input_shape):
- X = tf.random.uniform([int(input_shape[-1]),self.num_outputs],minval=0,maxval=1,dtype=tf.dtypes.float32,)
- k = tf.Variable(X, name="kernel")
- self.kernel = 0.5 * (k+tf.transpose(k))
- def call(self, input):
- return tf.matmul(input, self.kernel)
- layer = MyDenseLayer(5)
- print(layer(tf.ones([3, 5])))
- print(layer.trainable_variables)
- print(layer.trainable_variables)
- []
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement