torch tensor

Torch Tensor是PyTorch中的一个核心概念，它是一种多维数组，类似于NumPy中的ndarray。与NumPy数组不同的是，Torch Tensor可以在GPU上运行，可以利用GPU的并行计算能力进行高效的张量计算。同时，Torch Tensor还支持自动求导功能，可以方便地进行深度学习模型的训练。Torch Tensor的使用方式与NumPy数组类似，可以使用类似于数组的索引、切片等方式进行操作。以下是一个简单的创建和操作Torch Tensor的示例：

import torch

# 创建一个2x3的Tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 输出Tensor的形状
print(x.shape)

# 输出Tensor中的第一行
print(x[0])

# 将Tensor转换为NumPy数组
y = x.numpy()

# 将NumPy数组转换为Tensor
z = torch.from_numpy(y)