Sam Gross
f2d7e94948
Use torch.Size for Tensor sizes and tuple for strides
...
See issue #20
The torch.Size class is a tuple subclass which distinguishes sizes from
other tuples so that torch.Tensor(size) is interpreted as size instead
of data.
2016-10-28 19:37:09 +02:00
Sam Gross
ad2d413c0b
Add C++ bindings for cuDNN ( #167 )
...
The Python ctypes bindings overhead was high enough that it slowed down
multi-gpu training when using 4+ Maxwell GPUs.
2016-10-26 19:51:48 -04:00
Adam Paszke
f137c0c05a
Improve error messages of stateless functions
2016-10-24 22:29:43 +02:00
Adam Paszke
0325e2f646
Major autograd refactor
...
Improves autograd performance by more than 2x and fixes a couple
of bugs. All core functions have been moved to C.
2016-10-13 17:17:49 -07:00
Adam Paszke
e71204b52f
Improve error messages in storage and tensor C functions
2016-09-23 17:17:35 -07:00
Adam Paszke
06ab3f962f
Refactor _C extension to export some utilities
2016-09-21 08:36:54 -07:00
Adam Paszke
491fbfdc8c
Improve error messages of tensor methods
2016-09-15 18:49:20 -07:00
Adam Paszke
f9d186d33a
Add initial version of multiprocessing module
2016-08-31 19:46:08 -07:00
Adam Paszke
ef7364b80e
Fix Python 2.7 compatibility
2016-08-12 18:26:10 -07:00
Adam Paszke
bc7bd7a8b3
Add unit tests and fix detected bugs
2016-07-21 13:46:59 -04:00
Adam Paszke
2b53cce79f
Minor improvements in utils
2016-06-25 22:55:04 +02:00
Adam Paszke
523a6670f4
Improve cwrap error handling and fix memory leaks on error
2016-06-21 01:22:59 +02:00
Adam Paszke
4f66ea42af
Add random-related Tensor methods
2016-06-18 21:36:10 +02:00
Adam Paszke
0eb2b9e756
Add more Tensor and Storage methods
2016-06-15 23:03:47 +02:00
Adam Paszke
a9282edf79
Add THPPointer and more Tensor methods
2016-06-13 13:26:00 +02:00
Adam Paszke
60f9834ac6
Add more Tensor methods
2016-06-10 00:04:24 +02:00
Soumith Chintala
5ee3358a92
python 2 support
2016-06-08 19:14:57 -04:00
Adam Paszke
c3b3df9f22
Add utilities and clenup Tensor wrappers
2016-05-06 15:04:57 +02:00