def METHOD_NAME(self, inputs, mask=None):
    x, a = inputs
    mlp_out = self.mlp(x)
    output = mlp_out
    for _ in range(self.propagations):
        output = (1 - self.alpha) * ops.modal_dot(a, output) + self.alpha * mlp_out
    if mask is not None:
        output *= mask[0]
    output = self.activation(output)
    return output