facebookresearch/torchdim

Named tensors with first-class dimensions for PyTorch

Jupyter NotebookC++PythonC
This is stars and forks stats for /facebookresearch/torchdim repository. As of 08 May, 2024 this repository has 291 stars and 12 forks.

Named Tensors using First-class Dimensions in PyTorch -- Zachary DeVito @Zachary_DeVito An implementation of named tensors with the functionality of einsum , batching (vmap, xmap), and tensor indexing by adding dimension objects to PyTorch. The tensor input to a resnet might have the shape [8, 3, 224, 224] but informally we think of those dimensions as 'batch', 'channel', 'width', and 'height'. Eventhough 'width' and 'height' have the same size we still think of them as separate dimensions, and if...
Read on GithubGithub Stats Page
repotechsstarsweeklyforksweekly
GokuMohandas/mlops-courseJupyter NotebookPythonOther2.5k04450
dfinity/interface-specJavaScriptCSS310200
tobbelobb/hangprinterOpenSCADMakefileGAP4550990
floranderson/ProgramacionIPapyrusC0010
nettlep/magicSwiftC++Other8220300
devicekit/DeviceKitSwiftPythonRuby4.2k+5418+1
italia/design-drupal-themeTwigPHPSCSS7020
FrenzyExists/aquarium-vimVim ScriptShellC256+2130
vesoft-inc/nebulaC++GherkinPython9.5k+271.1k+5
stoneatom/stonedbC++CJavaScript825+21430