Method forward

method/blocks.py:220–246

(self, source: Tensor, pos_map: Tensor = None)

        self.depth = depth

    def forward(self, source: Tensor, pos_map: Tensor = None) -> Tensor:
        x = source

        x = self.norm1(x)

        x = self.cat(x, pos_map)
        x = self.conv1(x)
        x = self.activation(x)

        att = self.attention(x)

        x = x * att
        x = self.conv2(x)

        x = self.dropout1(x)

        y = source + x * self.beta

        x = self.conv3(self.norm2(y))
        x = self.activation(x)
        x = self.conv4(x)

        x = self.dropout2(x)

        ret = y + x * self.gamma

        return ret

class FeedForwardNetwork(nn.Module):
    def __init__(