pytorch
https://github.com/pytorch/pytorch
Python
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported920 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- Advanced indexing with uint8 tensor versus int64 tensor is inconsistent
- Define portable M_PI replacement, use it instead of non-standard M_PI in math.h
- Class based Sampler for Class Incremental/Continual Learning research
- torch.distributions.Binomial.sample() uses a massive amount of memory
- RuntimeError: Given input size: (2048x1x1). Calculated output size: (2048x-5x-5). Output size is too small at /pytorch/aten/src/THNN/generic/SpatialAveragePooling.c:48
- torch.nn.threshold cannot accept tensor as a threshold
- [FR] [RFC] add Sequential.append & .extend
- [JIT] traced model with optimization shows no performance improvement
- Performance difference between 0.4.1 and 1.1.0
- Performance issue master (a25b79531)
- Docs
- Python not yet supported