torch.tensor()
时间: 2023-08-03 09:09:10 浏览: 153
one hot编码:`torch.Tensor.scatter_()`函数用法详解
`torch.tensor()`是PyTorch中用于创建张量(tensor)的函数。
张量是PyTorch中的基本数据结构,类似于多维数组。`torch.tensor()`函数接受一个数据参数,并返回一个新的张量对象。
该函数有几种用法,可以根据传递的参数类型进行不同的操作:
1. 传递一个Python的列表、元组或NumPy数组作为参数,可以创建一个与输入数据相同的张量。
例如:`x = torch.tensor([1, 2, 3])`,创建了一个一维张量 `[1, 2, 3]`。
2. 传递一个标量值作为参数,可以创建一个只包含一个元素的张量。
例如:`x = torch.tensor(5)`,创建了一个包含单个元素5的张量。
3. 传递一个形状(shape)参数作为参数,可以创建一个指定形状的空张量。
例如:`x = torch.tensor((3, 4))`,创建了一个形状为`(3, 4)`的空张量。
`torch.tensor()`函数还可以接收一些可选参数,如数据类型(dtype)、设备(device)等,用于指定张量的属性。
总之,`torch.tensor()`函数是用于创建张量的便捷函数,根据不同的参数类型可以创建不同形状和类型的张量。
阅读全文