torch.tensor()用法
时间: 2024-02-25 17:38:57 浏览: 103
one hot编码:`torch.Tensor.scatter_()`函数用法详解
torch.tensor()是PyTorch框架中用于创建张量的函数,它可以从Python列表、元组、数值等数据类型中创建张量。可以指定dtype(数据类型)、device(计算设备)和requires_grad(是否需要计算梯度)等参数。例如,可以使用以下代码创建一个形状为(2,3)、数据类型为float32并保存在CPU上的张量:
import torch
tensor = torch.tensor([[1,2,3],[4,5,6]], dtype=torch.float32, device='cpu')
print(tensor)
阅读全文