Method forward

SAC-robot-navigation-CL/SAC/SAC.py:33–35

(self, x)

        self.register_buffer('pe', pe)

    def forward(self, x):
        x = x + self.pe[:x.size(0), :]
        return x

class EncoderBlock(nn.Module):
    def __init__(self, embed_dim, n_heads, dropout):