From bf70bcf41f85a65c2729e14e0742cd6e386f4eeb Mon Sep 17 00:00:00 2001 From: Kentaro Wada Date: Sun, 21 May 2017 05:30:07 +0900 Subject: [PATCH] Update types --- README.md | 8 +++++--- conversions.yaml | 16 ++++++++++------ 2 files changed, 15 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index 8d9dc10..23940e4 100644 --- a/README.md +++ b/README.md @@ -9,11 +9,13 @@ | Numpy | PyTorch | |:-------------|:---------------------| -| `np.uint8` | `torch.ByteTensor` | -| `np.int32` | `torch.IntTensor` | -| `np.int64` | `torch.LongTensor` | | `np.float32` | `torch.FloatTensor` | | `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 diff --git a/conversions.yaml b/conversions.yaml index 1592fef..cdef6b9 100644 --- a/conversions.yaml +++ b/conversions.yaml @@ -1,14 +1,18 @@ types: - - numpy: np.uint8 - pytorch: torch.ByteTensor - - numpy: np.int32 - pytorch: torch.IntTensor - - numpy: np.int64 - pytorch: torch.LongTensor - numpy: np.float32 pytorch: torch.FloatTensor - numpy: np.float64 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: - numpy: np.empty((2, 3))