torch.from_numpy(ndarray) 功能及举例

功能：torch.from_numpy(ndarray) → Tensor，即 从numpy.ndarray创建一个张量。

说明：返回的张量和ndarray共享同一内存。对张量的修改将反映在ndarray中，反之亦然。返回的张量是不能调整大小的。

举例：

>>> import torch

>>> import numpy as np

>>> a = np.array([1, 2, 3])

>>> t = torch.from_numpy(a)

>>> t
tensor([1, 2, 3], dtype=torch.int32)

>>> t[1]=1000

>>> t
tensor([   1, 1000,    3], dtype=torch.int32)

>>> a
array([   1, 1000,    3])

>>> a[0]=101

>>> a
array([ 101, 1000,    3])

>>> t
tensor([ 101, 1000,    3], dtype=torch.int32)

参考：https://pytorch.org/docs/stable/generated/torch.from_numpy.html?highlight=torch%20from_numpy