Code Monkey home page Code Monkey logo

di-treetensor's Issues

PyTorch OP Doc List

P0

  • cpu
  • cuda
  • to
  • torch.zeros_like
  • torch.randn_like
  • torch.ones_like
  • torch.zeros
  • torch.randn
  • torch.randint
  • torch.ones
  • cat
  • reshape
  • split
  • squeeze
  • stack
  • unsqueeze
  • where
  • abs
  • add
  • clamp
  • div
  • exp
  • log
  • sqrt
  • sub
  • sigmoid
  • pow
  • mul
  • argmax
  • argmin
  • all
  • any
  • max
  • min
  • dist
  • mean
  • std
  • sum
  • eq
  • ge
  • gt
  • le
  • lt
  • ne
  • clone
  • dot
  • matmul
  • mm

P1

  • numel
  • torch.randint_like
  • torch.full_like
  • torch.empty_like
  • torch.full
  • torch.empty
  • chunk
  • gather
  • index_select
  • masked_select
  • scatter
  • tile
  • unbind
  • ceil
  • exp2
  • floor
  • log10
  • log2
  • neg
  • round
  • sign
  • bmm

P2

  • logsumexp
  • median
  • norm
  • prod
  • unique
  • argsort
  • isfinite
  • isinf
  • isnan
  • sort
  • topk
  • cdist
  • flip
  • addbmm
  • addmm

Special OP List

Reduce Result OP

  • all
  • any

Input/Output Tuple OP

  • stack
  • cat
  • split

High Precision OP

  • mean
  • std
  • sum

PyTorch OP List(P0)

reference: https://pytorch.org/docs/1.8.0/torch.html

common

  • numel
  • cpu
  • cuda
  • to

Creation Ops

  • torch.zeros_like
  • torch.randn_like
  • torch.randint_like
  • torch.ones_like
  • torch.full_like
  • torch.empty_like
  • torch.zeros
  • torch.randn
  • torch.randint
  • torch.ones
  • torch.full
  • torch.empty

Indexing, Slicing, Joining, Mutating Ops

  • cat
  • chunk
  • gather
  • index_select
  • masked_select
  • reshape
  • scatter
  • split
  • squeeze
  • stack
  • tile
  • unbind
  • unsqueeze
  • where

Math Ops

Pointwise Ops
  • add
  • sub
  • mul
  • div
  • pow
  • neg
  • abs
  • sign
  • floor
  • ceil
  • round
  • sigmoid
  • clamp
  • exp
  • exp2
  • sqrt
  • log
  • log10
  • log2
Reduction Ops
  • argmax
  • argmin
  • all
  • any
  • max
  • min
  • dist
  • logsumexp
  • mean
  • median
  • norm
  • prod
  • std
  • sum
  • unique
Comparison Ops
  • argsort
  • eq
  • ge
  • gt
  • isfinite
  • isinf
  • isnan
  • le
  • lt
  • ne
  • sort
  • topk
Other Ops
  • cdist
  • clone
  • flip

BLAS and LAPACK Ops

  • addbmm
  • addmm
  • bmm
  • dot
  • matmul
  • mm

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.