本文简要介绍python语言中 torch.tensor
的用法。
用法:
torch.tensor(data, *, dtype=None, device=None, requires_grad=False, pin_memory=False) → Tensor
data(array_like) -张量的初始数据。可以是列表、元组、NumPy
ndarray
、标量和其他类型。dtype(
torch.dtype
, 可选的) -返回张量的所需数据类型。默认值:如果None
,则从data
推断数据类型。device(
torch.device
, 可选的) -返回张量的所需设备。默认值:如果None
,使用当前设备作为默认张量类型(参见torch.set_default_tensor_type()
)。device
将是 CPU 张量类型的 CPU 和 CUDA 张量类型的当前 CUDA 设备。requires_grad(bool,可选的) -如果 autograd 应该在返回的张量上记录操作。默认值:
False
。pin_memory(bool,可选的) -如果设置,返回的张量将被分配到固定内存中。仅适用于 CPU 张量。默认值:
False
。
用
data
构造一个张量。警告
torch.tensor()
始终复制data
。如果您有张量data
并希望避免复制,请使用torch.Tensor.requires_grad_()
或torch.Tensor.detach()
。如果您有 NumPyndarray
并希望避免复制,请使用torch.as_tensor()
。警告
当 data 是张量
x
时,torch.tensor()
从传递的任何内容中读取“数据”,并构造一个叶变量。因此torch.tensor(x)
等效于x.clone().detach()
且torch.tensor(x, requires_grad=True)
等效于x.clone().detach().requires_grad_(True)
。建议使用clone()
和detach()
的等效项。例子:
>>> torch.tensor([[0.1, 1.2], [2.2, 3.1], [4.9, 5.2]]) tensor([[ 0.1000, 1.2000], [ 2.2000, 3.1000], [ 4.9000, 5.2000]]) >>> torch.tensor([0, 1]) # Type inference on data tensor([ 0, 1]) >>> torch.tensor([[0.11111, 0.222222, 0.3333333]], ... dtype=torch.float64, ... device=torch.device('cuda:0')) # creates a torch.cuda.DoubleTensor tensor([[ 0.1111, 0.2222, 0.3333]], dtype=torch.float64, device='cuda:0') >>> torch.tensor(3.14159) # Create a scalar (zero-dimensional tensor) tensor(3.1416) >>> torch.tensor([]) # Create an empty tensor (of size (0,)) tensor([])
参数:
关键字参数:
相关用法
- Python PyTorch tensorinv用法及代码示例
- Python PyTorch tensorsolve用法及代码示例
- Python PyTorch tensor_split用法及代码示例
- Python PyTorch tensordot用法及代码示例
- Python PyTorch trunc用法及代码示例
- Python PyTorch triu_indices用法及代码示例
- Python PyTorch triangular_solve用法及代码示例
- Python PyTorch to_map_style_dataset用法及代码示例
- Python PyTorch trace_module用法及代码示例
- Python PyTorch topk用法及代码示例
- Python PyTorch tile用法及代码示例
- Python PyTorch tanh用法及代码示例
- Python PyTorch transpose用法及代码示例
- Python PyTorch take_along_dim用法及代码示例
- Python PyTorch t用法及代码示例
- Python PyTorch take用法及代码示例
- Python PyTorch trapezoid用法及代码示例
- Python PyTorch tan用法及代码示例
- Python PyTorch tril_indices用法及代码示例
- Python PyTorch trace用法及代码示例
- Python PyTorch tril用法及代码示例
- Python PyTorch triu用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.tensor。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。