分类目录:《深入浅出Pytorch函数》总目录
从numpy.ndarray
创建一个张量。返回的张量和numpy.ndarray
共用内存,对张量的修改将反映在numpy.ndarray
,反之亦然。返回的张量不可调整大小。
语法
torch.from_numpy(ndarray)
实例
>>> a = numpy.array([1, 2, 3])
>>> t = torch.from_numpy(a)
>>> t
tensor([ 1, 2, 3])
>>> t[0] = -1
>>> a
array([-1, 2, 3])