分类目录:《深入浅出Pytorch函数》总目录
相关文章:
· 深入浅出TensorFlow2函数——tf.constant
· 深入浅出Pytorch函数——torch.tensor
· 深入浅出Pytorch函数——torch.as_tensor
· 深入浅出Pytorch函数——torch.Tensor
· 深入浅出PaddlePaddle函数——paddle.to_tensor文章来源:https://www.toymoban.com/news/detail-449671.html
将数据转换为张量,共享数据并尽可能保留自动微分的历史。如果数据已经是具有所需的数据类型和指定设备的张量,则返回数据本身,但是如果数据是具有不同数据类型或设备类型的张量,则像使用data.to(dtype=dtype, device=device)
复制它。文章来源地址https://www.toymoban.com/news/detail-449671.html
语法
torch.as_tensor(data, dtype=None, device=None)
参数
-
data
:张量的初始数据。可以是列表、元组、numpy.ndarray
、标量及其他类型 -
dtype
:[torch.dtype
, 可选] 返回所需张量的数据类型。如果为None
,则推断数据类型数据 -
device
:[torch.device
, 可选] 构造张量的设备。如果为None
且数据是张量,则使用数据的设备。如果为None
且数据不是张量,则结果张量在CPU上构建。
实例
>>> a = numpy.array([1, 2, 3])
>>> t = torch.as_tensor(a)
>>> t
tensor([ 1, 2, 3])
>>> t[0] = -1
>>> a
array([-1, 2, 3])
>>> a = numpy.array([1, 2, 3])
>>> t = torch.as_tensor(a, device=torch.device('cuda'))
>>> t
tensor([ 1, 2, 3])
>>> t[0] = -1
>>> a
array([1, 2, 3])
到了这里,关于深入浅出Pytorch函数——torch.as_tensor的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!