Skip to main content

Quality of life improvements for pytorch.

Project description

fix_torch

Quality of Life improvements for pytorch

Basically a combination of monkey-patches and additional functionality which is missing from pytorch. Use at your own peril :o

Examples

import fix_torch
import torch

x = torch.randn(13,127,12,3)
y = torch.randn(13,127,3,17)
v = torch.randn(8,4,4)

z = x@y # do bmm without torch being annoying about it
tr = torch.trace(v) # do trace on batched matrices

w = x.reshape(...,4,3,3)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

fix_torch-0.0.3.tar.gz (2.8 kB view details)

Uploaded Source

Built Distribution

fix_torch-0.0.3-py3-none-any.whl (2.5 kB view details)

Uploaded Python 3

File details

Details for the file fix_torch-0.0.3.tar.gz.

File metadata

  • Download URL: fix_torch-0.0.3.tar.gz
  • Upload date:
  • Size: 2.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.1 importlib_metadata/4.6.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.61.1 CPython/3.9.5

File hashes

Hashes for fix_torch-0.0.3.tar.gz
Algorithm Hash digest
SHA256 374ff16f284e9646e3e8e704cee2b9a10c8dea77e4f8818aba6b057a9a9ce0c1
MD5 4cf772d641eed851623dad56c6b230c8
BLAKE2b-256 ea20d3de2abe63dc98185eb4766e1f8fea4ebe2d3b88ce33b049f2cf0de083c7

See more details on using hashes here.

File details

Details for the file fix_torch-0.0.3-py3-none-any.whl.

File metadata

  • Download URL: fix_torch-0.0.3-py3-none-any.whl
  • Upload date:
  • Size: 2.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.4.1 importlib_metadata/4.6.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.61.1 CPython/3.9.5

File hashes

Hashes for fix_torch-0.0.3-py3-none-any.whl
Algorithm Hash digest
SHA256 420de64999a17ba05941c27c0a39d43ce65033019f6e4d9fded8f73ddcf5110e
MD5 776d7773337ad9f4b19d18397fca3f6d
BLAKE2b-256 71a66ae06f29f56ddd665f5e81f5149c9b93ca9846eb97e6a47cd32414e9a9f7

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page