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 Lerer
86288265ad
Adding rnn cell library
2016-10-23 20:23:48 -07:00
Adam Paszke
7991125293
Improve error messages
2016-10-08 20:37:40 -07:00
Adam Paszke
a33c9bd774
Improve argument matching in invalidArguments
2016-09-29 12:16:19 -07:00
Adam Paszke
0be5031a93
Pretty print type mismatches in error messages
2016-09-25 12:26:00 -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
e9f9fd3727
Major refactor
2016-08-10 09:24:53 -07:00
Adam Paszke
1d763810ba
Fix optional argument resolution in cwrap
2016-07-19 10:52:52 -04:00
Adam Paszke
3a44259b32
Add support for CUDA
2016-07-19 10:45: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