Method forward

encoders/dinov2/layers/layer_scale.py:26–27

(self, x: Tensor)

        self.gamma = nn.Parameter(init_values * torch.ones(dim))

    def forward(self, x: Tensor) -> Tensor:
        return x.mul_(self.gamma) if self.inplace else x * self.gamma