class MyFunction(torch.autograd.Function):
        """Some Information about MyFunction"""
      
        @staticmethod
        def forward(ctx, input):
      
          return
      
        @staticmethod
        def backward(ctx, grad_output)
      
          return
      Python language logo
      your recipe card header background

      PyTorch Autograd Function

      Pytorch Public Recipes

      Library: pytorch

      Shortcut: pytorch.function

      0 Comments

        Add Comment

        Log in to add a comment

        Codiga - All rights reserved 2022.