将PyTorch Tensor转换为整数类型

简介：在PyTorch中，可以使用`.long()`或`.int()`方法将Tensor转换为整数类型。这些方法将Tensor中的元素转换为32位或64位整数，具体取决于平台和数据类型。以下是一个示例代码，演示如何将Tensor转换为整数类型：

import torch
# 创建一个浮点数类型的Tensor
x = torch.tensor([1.5, 2.7, 3.2], dtype=torch.float32)
# 将Tensor转换为整数类型
x_int = x.long()
# 输出转换后的Tensor
print(x_int)