Method forward

advanced_source/neural_style_tutorial.py:243–246

(self, input)

        self.target = gram_matrix(target_feature).detach()

    def forward(self, input):
        G = gram_matrix(input)
        self.loss = F.mse_loss(G, self.target)
        return input


######################################################################