将 PyTorch 张量从 float32 转换为 long 类型
要将一个 PyTorch 的 float32 类型张量向量转换为 long 类型,可以使用 torch.long() 方法。
以下是一个示例代码:
import torch
# 创建一个 float32 类型的张量向量
float_tensor = torch.tensor([1.0, 2.0, 3.0, 4.0], dtype=torch.float32)
# 将 float32 类型的张量向量转换为 long 类型
long_tensor = float_tensor.long()
# 打印转换后的 long 类型的张量向量
print(long_tensor)
输出结果为:
tensor([1, 2, 3, 4])
通过调用 float_tensor.long() 方法,可以将 float32 类型的张量向量转换为 long 类型。
原文地址: https://www.cveoy.top/t/topic/iNyG 著作权归作者所有。请勿转载和采集!