def call()

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)