def forward(self, x):
    inp = F.relu(self.all_mutual(x))
    m1 = F.relu(self.m1(inp)).squeeze()
    return m1
def forward(self, x):
    inp = F.relu(self.all_mutual(x))
    m1 = F.relu(self.m1(inp)).squeeze()
    return m1