torch.tensor的dtype类型使用范例

torch.tensor是PyTorch中的一个重要数据结构，用于创建多维数组（张量）。dtype参数用于指定张量的数据类型。以下是一些使用不同dtype类型的torch.tensor的范例：

1. 使用默认的dtype（float32）创建一个张量：

   import torch
   
   x = torch.tensor([1, 2, 3])
   print(x.dtype)  # 输出为torch.float32

2. 使用int64的dtype创建一个张量：

   import torch
   
   x = torch.tensor([1, 2, 3], dtype=torch.int64)
   print(x.dtype)  # 输出为torch.int64

3. 使用bool的dtype创建一个布尔型张量：

   import torch
   
   x = torch.tensor([True, False, True], dtype=torch.bool)
   print(x.dtype)  # 输出为torch.bool

4. 使用float16的dtype创建一个半精度浮点型张量：

   import torch
   
   x = torch.tensor([1.0, 2.0, 3.0], dtype=torch.float16)
   print(x.dtype)  # 输出为torch.float16

5. 使用complex128的dtype创建一个复数型张量：

   import torch
   
   x = torch.tensor([1+2j, 3+4j], dtype=torch.complex128)
   print(x.dtype)  # 输出为torch.complex128