in train/CustomModel.py [0:0]
def call(self, x):
e = self.activation(K.dot(x,self.W)+self.b)
a = K.softmax(e, axis=1)
output = x*a
if self.return_sequences:
return output
return K.sum(output, axis=1)