Adam Paszke
7c5014d803
Add torch.split, torch.chunk and change default dim of cat to 0
2016-12-31 16:25:39 -05:00
Adam Paszke
62ac1b4bdd
Implement missing cases of __matmul__
2016-12-31 16:25:39 -05:00
Adam Paszke
0633c08ec9
Add is_shared() method for storages and tensors
2016-12-31 16:25:39 -05:00
Sergey Zagoruyko
62af45d99f
Basic functional interface ( #354 )
2016-12-29 22:53:57 +01:00
Sam Gross
24af02154c
Use ForkingPickler for sharing tensor/storages across processes ( #344 )
...
This hooks into the (internal) ForkingPickler class in multiprocessing
to reduce tensors, storages, and CUDA events instead of our queue from
joblib. This makes it easier to use the standard multiprocessing classes
in later versions of Python.
This also exposes:
- Tensor/Storage.share_memory_()
- Module.share_memory()
These methods move the CPU tensors and storages to shared memory. If
you're using the "fork" method of multiprocessing, these objects can be
directly inherited instead of serialized through a queue.
2016-12-28 20:34:23 -05:00
Adam Paszke
e27bb3e993
Minor fixes
2016-12-28 18:15:17 +01:00
Adam Paszke
cd82b2b869
Implement comparison and logical operators for tensors
2016-12-28 00:04:08 +01:00
soumith
a215e000e9
fix for out of place tests and for non standard I/O pipes
2016-12-20 16:13:24 -08:00
Sam Gross
1af9a9637f
Refactor copy and release GIL during copy ( #286 )
2016-12-11 21:54:58 +01:00
Adam Paszke
2fd78112ab
Add half copy/conversions
2016-11-17 14:34:33 -08:00
Soumith Chintala
44874542c8
fix printing in console ( #208 )
2016-11-08 13:42:26 -05:00
Zeming Lin
86e42ba291
Adding truncated tensor printing ( #202 )
...
* Adding truncated tensor printing
2016-11-08 10:05:30 -05:00
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
ee14cf9438
Add support for pinned memory: ( #127 )
...
torch.Storage/Tensor.pin_memory()
torch.Storage/Tensor.is_pinned()
2016-10-15 18:38:26 -04:00
Sam Gross
0391bbb376
Fix view_as and view for empty tensors ( #128 )
2016-10-15 18:33:05 -04: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
Sam Gross
cb5d4e836f
Lazy load CUDA and THNN modules ( #64 )
2016-09-28 19:29:53 -04:00
Adam Paszke
1828e7c42f
Add async CUDA copy
2016-09-27 15:12:48 -07:00
Adam Paszke
06ab3f962f
Refactor _C extension to export some utilities
2016-09-21 08:36:54 -07:00
Adam Paszke
8fdec15a55
Codemod to remove camel case method naming
2016-09-20 08:40:28 -07:00
