def forward(self, x):
    x = self.Edge_Net_head(x)
    res = x
    MSRB_out = []
    for i in range(self.n_blocks):
        x = self.Edge_Net_body[i](x)
        MSRB_out.append(x)
    MSRB_out.append(res)
    res = torch.cat(MSRB_out, 1)
    x = self.Edge_Net_tail(res)
    return x
def forward(self, x):
    x = self.Edge_Net_head(x)
    res = x
    MSRB_out = []
    for i in range(self.n_blocks):
        x = self.Edge_Net_body[i](x)
        MSRB_out.append(x)
    MSRB_out.append(res)
    res = torch.cat(MSRB_out, 1)
    x = self.Edge_Net_tail(res)
    return x