wassname/pytorch-for-numpy-users
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-for-numpy-users.git synced 2026-06-27 16:10:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update types

Browse Source
This commit is contained in:
Kentaro Wada
2017-05-21 05:30:07 +09:00
parent 1c6a9e5474
commit bf70bcf41f
2 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+5 -3
View File
@@ -9,11 +9,13 @@
| Numpy | PyTorch | | Numpy | PyTorch |
|:-------------|:---------------------| |:-------------|:---------------------|
| `np.uint8` | `torch.ByteTensor` |
| `np.int32` | `torch.IntTensor` |
| `np.int64` | `torch.LongTensor` |
| `np.float32` | `torch.FloatTensor` | | `np.float32` | `torch.FloatTensor` |
| `np.float64` | `torch.DoubleTensor` | | `np.float64` | `torch.DoubleTensor` |
| `np.int8` | `torch.CharTensor` |
| `np.uint8` | `torch.ByteTensor` |
| `np.int16` | `torch.ShortTensor` |
| `np.int32` | `torch.IntTensor` |
| `np.int64` | `torch.LongTensor` |
## Constructors ## Constructors
conversions.yaml
+10 -6
View File
@@ -1,14 +1,18 @@
types: types:
- numpy: np.uint8
pytorch: torch.ByteTensor
- numpy: np.int32
pytorch: torch.IntTensor
- numpy: np.int64
pytorch: torch.LongTensor
- numpy: np.float32 - numpy: np.float32
pytorch: torch.FloatTensor pytorch: torch.FloatTensor
- numpy: np.float64 - numpy: np.float64
pytorch: torch.DoubleTensor pytorch: torch.DoubleTensor
- numpy: np.int8
pytorch: torch.CharTensor
- numpy: np.uint8
pytorch: torch.ByteTensor
- numpy: np.int16
pytorch: torch.ShortTensor
- numpy: np.int32
pytorch: torch.IntTensor
- numpy: np.int64
pytorch: torch.LongTensor
constructors: constructors:
- numpy: np.empty((2, 3)) - numpy: np.empty((2, 3))