在PyTorch中，我们经常需要把Tensor转换成Int，以便于进行整数运算或者把Tensor输入到仅接受整数类型的函数中。本文将详细介绍如何使用PyTorch将Tensor转换成Int，并阐述其中的重点词汇和短语。
import torch# 创建一个float类型的Tensortensor_float = torch.tensor([1.2, 2.3, 3.4])print(f"原始Tensor数据类型: {tensor_float.dtype}")# 转换为long类型tensor_long = tensor_float.long()print(f"转换为long类型: {tensor_long.dtype}")# 转换为int类型tensor_int = tensor_float.int()print(f"转换为int类型: {tensor_int.dtype}")
在这个例子中，我们首先创建了一个float类型的Tensor，然后用
.long()
和
.int()
方法分别将它转换成long和int类型的Tensor。可以看到，
.long()
和
.int()
方法并不改变Tensor中的数值，只是改变了它们的类型。
