Files changed (1) hide show
  1. tkan.py +0 -18
tkan.py DELETED
@@ -1,18 +0,0 @@
1
- import tensorflow as tf
2
- from tensorflow.keras import layers
3
-
4
- class TKAN(tf.keras.layers.Layer):
5
- def __init__(self, units, **kwargs):
6
- super(TKAN, self).__init__(**kwargs)
7
- self.units = units
8
- self.dense1 = layers.Dense(units, activation="relu")
9
- self.dense2 = layers.Dense(units, activation="relu")
10
-
11
- def call(self, inputs):
12
- x = self.dense1(inputs)
13
- return self.dense2(x)
14
-
15
- def get_config(self):
16
- config = super(TKAN, self).get_config()
17
- config.update({"units": self.units})
18
- return config