PyTorch Autograd Function

    0

    0

    lucycodes42

    Pytorch Public Recipes

    Library: pytorch

    Shortcut: pytorch.function

    class MyFunction(torch.autograd.Function):
      """Some Information about MyFunction"""
    
      @staticmethod
      def forward(ctx, input):
    
        return
    
      @staticmethod
      def backward(ctx, grad_output)
    
        return
    Codiga Logo
    Codiga Hub
    • Rulesets
    • Playground
    • Snippets
    • Cookbooks
    soc-2 icon

    We are SOC-2 Compliance Certified

    G2 high performer medal

    Codiga – All rights reserved 2022.