1 生成张量
1.1 使用torch.tensor()或者torch.Tensor()来生成张量。
两者主要区别为
torch.Tensor(data)
:直接调用Tensor
构造函数会默认创建一个浮点型张量 (torch.float32
)。即使输入的数据是整数列表或元组,生成的张量元素也会被转换为浮点数。torch.tensor(data)
:此函数会根据输入数据的类型推断出张量的类型。如果输入的是整数列表,则生成的张量将具有整数类型(例如torch.int64
),而浮点数列表则会生成浮点类型的张量。