fixed to use tensorflow

Eloypripan · Jan 3, 2018 · b7e0f3e · b7e0f3e
1 parent 2ebd81c
commit b7e0f3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rbflayer.py b/rbflayer.py
@@ -59,7 +59,7 @@ def build(self, input_shape):
                                        initializer=self.initializer,
                                        trainable=True)
         self.betas = self.add_weight(name='betas',
-                                     shape=(self.output_dim),
+                                     shape=(self.output_dim,),
                                      initializer=Constant(value=self.init_betas),
                                      #initializer='ones',
                                      trainable=True)
@@ -69,7 +69,7 @@ def build(self, input_shape):
     def call(self, x):
 
         C = K.expand_dims(self.centers)
-        H = (C-x.T).T
+        H = K.transpose(C-K.transpose(x))
         return K.exp( -self.betas * K.sum(H**2, axis=1))
 
         #C = self.centers[np.newaxis, :, :]