本教程将带你了解PyTorch中张量(Tensor)的基本概念与操作,适合初学者入门深度学习!

🛠️ 什么是张量?

张量是PyTorch中数据的核心表示形式,可视为多维数组。

  • 标量(0-D): 单个数值,如 5
  • 向量(1-D): 一维数组,如 [1, 2, 3]
  • 矩阵(2-D): 二维数组,如 [[1, 2], [3, 4]]
  • 高维张量: 三维及以上,常用于处理图像、文本等数据
PyTorch_Tensor

🧮 常见操作示例

import torch

# 创建张量
x = torch.tensor([1, 2, 3])  # 1-D
y = torch.tensor([[1, 2], [3, 4]])  # 2-D

# 基础运算
print(x + y)  # 张量加法
print(torch.mul(x, y))  # 张量乘法
print(torch.sum(y))  # 求和

🚀 应用场景

  • 数据存储: 用于保存模型参数或输入数据
  • 计算图构建: 张量操作自动记录计算过程,支持反向传播
  • GPU加速: 张量可迁移到GPU进行高效计算
Tensor_Operations

❓ 常见问题

  1. 如何查看张量维度?
    使用 .shape 属性,如 x.shape 返回 (3,)
  2. 如何将张量转为NumPy数组?
    通过 .numpy() 方法,但需注意内存共享机制

点击了解更多PyTorch基础内容