pytorch输出张量数据类型
时间: 2023-07-06 17:33:41 浏览: 108
pytorch中tensor张量数据类型的转化方式
5星 · 资源好评率100%
在 PyTorch 中,可以使用 `dtype` 属性获取张量的数据类型。PyTorch 支持多种数据类型,包括浮点数、整数、布尔值等。
示例代码如下:
```
import torch
# 创建一个浮点数张量
x = torch.FloatTensor([1.0, 2.0, 3.0])
print(x.dtype) # 输出 torch.float32
# 创建一个整数张量
y = torch.LongTensor([1, 2, 3])
print(y.dtype) # 输出 torch.int64
# 创建一个布尔值张量
z = torch.BoolTensor([True, False, True])
print(z.dtype) # 输出 torch.bool
```
在这个示例代码中,我们分别创建了一个浮点数张量、一个整数张量和一个布尔值张量,并使用 `dtype` 属性获取了它们的数据类型。输出结果分别为 `torch.float32`、`torch.int64` 和 `torch.bool`,分别对应浮点数类型、整数类型和布尔值类型。
阅读全文