Spaces:
Sleeping
Sleeping
Delete tkan.py
#1
by
dsid271 - opened
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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|