mirror of
https://github.com/wassname/pytorch-for-numpy-users.git
synced 2026-06-27 16:10:21 +08:00
Kentaro Wada
42 lines
943 B
YAML
42 lines
943 B
YAML
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
|
|
|
|
constructors:
|
|
- numpy: np.empty((2, 3))
|
|
pytorch: torch.Tensor(2, 3)
|
|
- numpy: np.empty_like(x)
|
|
pytorch: x.new(x.size()).type(x.type())
|
|
- numpy: np.eye
|
|
pytorch: torch.eye
|
|
- numpy: np.identity
|
|
pytorch: torch.eye
|
|
- numpy: np.ones
|
|
pytorch: torch.ones
|
|
- numpy: np.ones_like
|
|
pytorch: torch.ones(x.size()).type(x.type())
|
|
- numpy: np.zeros
|
|
pytorch: torch.zeros
|
|
- numpy: np.zeros_like
|
|
pytorch: torch.zeros(x.size()).type(x.type())
|
|
|
|
# - numpy: x.astype(np.int32)
|
|
# pytorch: x.type(torch.IntTensor)
|
|
#
|
|
# - numpy: y = x.copy()
|
|
# pytorch: y = x.clone()
|
|
#
|
|
# - numpy: x.shape
|
|
# pytorch: x.size()
|
|
#
|
|
# - numpy: x.size
|
|
# pytorch: x.nelement()
|