PyTorch Tensors 简介

PyTorch Tensors 是 PyTorch 库中用于存储和处理数据的结构。它们类似于 NumPy 的 ndarrays，但具有一些额外的功能，例如自动微分。

Tensors 的基本操作

以下是一些常见的 Tensors 操作：

• 创建 Tensor：使用 torch.tensor() 函数可以创建一个 Tensor。
• 索引和切片：与 NumPy 类似，可以使用索引和切片来访问 Tensors 的特定部分。
• 数学运算：Tensors 支持各种数学运算，例如加法、减法、乘法和除法。

示例

以下是一个简单的示例，演示如何创建和操作 Tensors：

import torch

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

# 打印 Tensor
print(tensor)

# 访问特定元素
print(tensor[0, 1])  # 输出 2

# 切片
print(tensor[0:1, 0:2])  # 输出 [[1 2]]

更多信息

要了解更多关于 Tensors 的信息，请访问 PyTorch Tensors 官方文档。

Tensor 示例