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
Release history Release notifications | RSS feed
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 hashes)
Built Distribution
Close
Hashes for fix_torch-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 420de64999a17ba05941c27c0a39d43ce65033019f6e4d9fded8f73ddcf5110e |
|
MD5 | 776d7773337ad9f4b19d18397fca3f6d |
|
BLAKE2b-256 | 71a66ae06f29f56ddd665f5e81f5149c9b93ca9846eb97e6a47cd32414e9a9f7 |