torch.LongTensor使用方法

torch.LongTensor 是 PyTorch 中的一个数据类型，用于表示包含整数（整型数据）的张量（tensor）。它是一种特定的张量类型，其中的元素都为整数类型，并且使用 64 位整数进行存储。

以下是创建和使用 torch.LongTensor 的示例：

import torch

# 创建一个包含整数的张量
data = [1, 2, 3, 4, 5]
tensor = torch.LongTensor(data)
print(tensor)
# 输出: tensor([1, 2, 3, 4, 5])

# 进行张量操作
result = tensor + 10
print(result)
# 输出: tensor([11, 12, 13, 14, 15])

# 访问张量中的元素
value = tensor[0]
print(value)
# 输出: tensor(1)

在这个示例中，我们首先通过传递一个包含整数的 Python 列表 data 来创建 torch.LongTensor。然后，我们可以对其执行各种张量操作，比如加法运算和索引取值。最后打印结果，可以看到张量中的整数值。

需要注意的是，torch.LongTensor 在 PyTorch 1.6 版本之后被弃用，推荐使用 torch.tensor 并指定 dtype=torch.long 来创建相同类型的张量。例如：

tensor = torch.tensor(data, dtype=torch.long)