百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT知识 > 正文

Pytorch学习-Day 3:张量高级操作(张量svd)

liuian 2025-05-08 19:40 34 浏览

以下是 PyTorch 第3天学习任务的示例代码,涵盖张量形状变换(view 和 reshape)、广播机制以及 GPU 加速的实现。代码中包含详细注释,帮助理解每个部分的功能,并完成任务:将张量移动到 GPU 并计算两个张量的矩阵乘法。

python

# 示例代码:PyTorch Day 3 - 张量高级操作

# 导入 PyTorch 库
import torch

# 检查 GPU 是否可用
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"当前设备: {device}")

# 1. 张量形状变换(view 和 reshape)
def tensor_shape_transform():
    # 创建一个 3x4 的张量
    x = torch.arange(12).float()  # 生成 0 到 11 的张量,转换为浮点型
    print("原始张量 x:")
    print(x)
    print(f"x 的形状: {x.shape}")  # torch.Size([12])

    # 使用 view 变换形状(要求元素总数不变)
    x_view = x.view(3, 4)  # 变换为 3x4
    print("\n使用 view 变换后的 x:")
    print(x_view)
    print(f"x_view 的形状: {x_view.shape}")  # torch.Size([3, 4])

    # 使用 reshape 变换形状(更灵活,可自动推导维度)
    x_reshape = x.reshape(2, -1)  # -1 表示自动计算此维度,变为 2x6
    print("\n使用 reshape 变换后的 x:")
    print(x_reshape)
    print(f"x_reshape 的形状: {x_reshape.shape}")  # torch.Size([2, 6])

# 2. 广播机制(Broadcasting)
def tensor_broadcasting():
    # 创建两个形状不同的张量
    a = torch.tensor([[1, 2, 3]])  # 形状: 1x3
    b = torch.tensor([[4], [5], [6]])  # 形状: 3x1
    print("\n张量 a:")
    print(a)
    print(f"a 的形状: {a.shape}")
    print("张量 b:")
    print(b)
    print(f"b 的形状: {b.shape}")

    # 广播机制自动扩展维度并进行逐元素相加
    c = a + b  # 结果形状为 3x3
    print("\n广播后 a + b 的结果:")
    print(c)
    print(f"c 的形状: {c.shape}")

# 3. GPU 加速 + 矩阵乘法任务
def matrix_multiplication_on_gpu():
    # 创建两个张量
    A = torch.randn(3, 4)  # 随机生成 3x4 矩阵
    B = torch.randn(4, 5)  # 随机生成 4x5 矩阵
    print("\n矩阵 A:")
    print(A)
    print("矩阵 B:")
    print(B)

    # 将张量移动到 GPU(如果可用)
    if torch.cuda.is_available():
        A = A.to(device)
        B = B.to(device)
        print(f"\n张量已移动到 {device}")

    # 计算矩阵乘法
    C = torch.matmul(A, B)  # 结果形状为 3x5
    print("\n矩阵乘法结果 C (A @ B):")
    print(C)
    print(f"C 的形状: {C.shape}")

# 主函数:依次运行所有任务
def main():
    print("=== 张量形状变换 ===")
    tensor_shape_transform()
    print("\n=== 广播机制 ===")
    tensor_broadcasting()
    print("\n=== GPU 加速与矩阵乘法 ===")
    matrix_multiplication_on_gpu()

if __name__ == "__main__":
    main()

"""
学习内容说明:
1. 张量形状变换:
   - view(): 改变张量形状,要求内存连续且元素总数不变。
   - reshape(): 更灵活的形状变换,支持非连续内存。

2. 广播机制:
   - 自动扩展张量维度以匹配操作(如加法),无需手动调整形状。
   - 规则:从尾部对齐维度,小维度扩展为大维度或补1。

3. GPU 加速:
   - 使用 torch.device 和 .to() 将张量移到 GPU。
   - torch.cuda.is_available() 检查 GPU 可用性。

任务完成:
- 将张量 A 和 B 移动到 GPU(如果可用)。
- 使用 torch.matmul() 计算矩阵乘法。
"""

代码说明

  1. 张量形状变换:
  2. view(3, 4): 将一维张量变换为 3x4 的二维张量。
  3. reshape(2, -1): 将张量变为 2 行,列数自动推导为 6。
  4. 广播机制:
  5. 两个张量 a (1x3) 和 b (3x1) 通过广播扩展为 3x3,然后逐元素相加。
  6. 展示了 PyTorch 如何自动处理维度不匹配的情况。
  7. GPU 加速与矩阵乘法:
  8. 检查 GPU 可用性并定义 device。
  9. 使用 .to(device) 将张量 A 和 B 移到 GPU。
  10. 使用 torch.matmul() 计算矩阵乘法,结果为 3x5 的矩阵。

运行要求

  • 安装 PyTorch: pip install torch(如果需要 GPU 支持,确保安装 CUDA 版本,例如 pip install torch torchvision -f https://download.pytorch.org/whl/cu117)。
  • 有 GPU 的环境(可选):如果没有 GPU,代码会自动在 CPU 上运行。

如何运行

  1. 保存代码为 pytorch_day3.py。
  2. 在终端运行:python pytorch_day3.py。
  3. 观察输出,验证形状变换、广播和矩阵乘法的结果。

输出示例(部分)

当前设备: cuda
=== 张量形状变换 ===
原始张量 x:
tensor([ 0.,  1.,  2.,  3.,  4.,  5.,  6.,  7.,  8.,  9., 10., 11.])
x 的形状: torch.Size([12])

使用 view 变换后的 x:
tensor([[ 0.,  1.,  2.,  3.],
        [ 4.,  5.,  6.,  7.],
        [ 8.,  9., 10., 11.]])
x_view 的形状: torch.Size([3, 4])

...
=== GPU 加速与矩阵乘法 ===
张量已移动到 cuda
矩阵乘法结果 C (A @ B):
tensor([[...], [...], [...]])  # 具体值因随机生成而异
C 的形状: torch.Size([3, 5])

这个代码完整实现了第3天的学习任务,并通过注释和输出展示了 PyTorch 的张量高级操作。建议参考 PyTorch 官方教程“Tensor Operations”进一步深入学习!

相关推荐

微信主页背景墙壁纸怎么设置
  • 微信主页背景墙壁纸怎么设置
  • 微信主页背景墙壁纸怎么设置
  • 微信主页背景墙壁纸怎么设置
  • 微信主页背景墙壁纸怎么设置
校园网wifi免认证软件(校园网统一身份认证平台)

这个不存在犯法不犯法的问题,也就是说学校的网络是给你便捷使用的,反正都是给你使用的,你如何登录都没有任何的关系,其次就是你自己办的网的话,你有权利随意的更改,没办网的话那你就用学校的。1这是不道德和...

如何查看windows激活密钥(查看windows激活密钥命令)

可以按照以下步骤查看Windows系统的激活密钥:1.首先打开命令提示符,可通过在搜索栏中输入"cmd",然后右键管理员身份打开。2.在打开的命令提示符窗口中输入指令:slmgr/d...

dlink路由器(dlink路由器无法连接网络)

设置D-Link无线路由器无线桥接的具体步骤如下:1、将电脑与路由器的任意lan口连接,打开浏览器输入192.168.1.1,进入路由器管理页面。点击lan口设置,将lan口ip改为192.168.2...

c5game开箱网(c5game开箱网是正规的吗)

苹果c5game开箱操作很简单,首先进入c5game网站,选择打开自己的背包,然后找到自己想要开箱的物品,点击开箱按钮即可。在开箱过程中,会弹出一个开箱界面,按照界面提示进行操作,等待开箱过程结束即可...

ps5官网(playstation 官网)

在官网买ps5需要玩家收到预购邀请才可以。索尼决定遴选出一批忠实玩家,率先向其提供PS5实机预定服务,数量有限,先到先得。玩家只需在PlayStation.com网站完成注册手续。若有幸等到预购邀请电...

笔记本添加打印机步骤(电脑添加打印机步骤)
  • 笔记本添加打印机步骤(电脑添加打印机步骤)
  • 笔记本添加打印机步骤(电脑添加打印机步骤)
  • 笔记本添加打印机步骤(电脑添加打印机步骤)
  • 笔记本添加打印机步骤(电脑添加打印机步骤)
电脑上dat文件用什么打开(电脑上dat文件怎么打开)

、打开电脑,找到“我的电脑”然后再打开硬盘C就可以看到相应的dat文件。2、硬盘C里面可以找到很多的dat文件,只是他们的文件拓展名不一样。3、然后在我的电脑当中输入“dat”就会弹出许多与dat相关...

win10 2004和1909哪个好用(win10版本1909好用还是2004好用)

可以下载升级包进行升级。

主题商店oppo免费下载(oppo主题商店7.0.0下载)

如果在oppo手机上面下载主题上练的话,我建议还是不要下载,因为手机上自带有主题商店,然后可以在自带的主题商店里面就能下载使用自己的主题,再一个就是我们下载的主题商店,也会占手机内存,也可以导致手机内...

win10系统 pe(win10系统 PID 1716)

E10都是表示系统缺水,给壁挂炉充上水压补足1.2bar即可解决。用户在使用前,首先应检查锅炉的水压表指针是否在规定范围内,说明书中规定的标准水压为1-1.2bar。但在实际使用过程中,由于暖气系统和...

极路由登录(极路由登录网址是什么)
  • 极路由登录(极路由登录网址是什么)
  • 极路由登录(极路由登录网址是什么)
  • 极路由登录(极路由登录网址是什么)
  • 极路由登录(极路由登录网址是什么)
电信官网登录入口(中国电信官网登录入口)

中国电信天翼网关一般情况下,登陆地址是http://192.168.0.1,用提供的用户名和密码就可以登陆。电信智能网关登录的方法就是要看电信光纤猫上背面的标签上的登录管理页面的网址以及登录的用户名和...

比迅雷好用的下载工具(比迅雷好用的下载app)

迅雷是一款中国大陆的著名下载工具,但由于其限速和广告等问题,许多用户可能会寻求更良心、更干净的替代品。以下是一些建议的下载工具:1.IDM(InternetDownloadManager):这是...

win2007是win7吗(windows 7是哪一年的)
  • win2007是win7吗(windows 7是哪一年的)
  • win2007是win7吗(windows 7是哪一年的)
  • win2007是win7吗(windows 7是哪一年的)
  • win2007是win7吗(windows 7是哪一年的)