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

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

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

以下是 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”进一步深入学习!

相关推荐

ie浏览器手机版官网下载(ie游览器手机版下载)

如果您在使用IE浏览器时遇到无法下载的问题,以下是一些常见的解决办法:1.清除浏览器缓存:打开IE浏览器,依次点击工具(齿轮图标)->Internet选项->常规选项->...

office2003属于什么软件(word2003属于什么软件)

是一套Office2003专业版的精简版,包含常用的Word、Excel、PowerPoint三个应用,使用者甚多。楼主如果有需要,请上电脑在本帖下载我的附件。我见过最多的,是2013或者以上的(因为...

电脑鼠标设置在哪里调(电脑鼠标在哪里去调)

电脑点击开始,在菜单中找到“控制面板”,点击“控制面板”进入,找到“鼠标”点击进入在打开的窗口中选择“指针”,选择指针样式,可点击浏览,找到文件夹下,查看哪些指针可选择。可按路径把喜欢的图标放进去找到...

ie浏览器怎么下载到电脑桌面

工具/材料:电脑1、首先在电脑桌面里找到这台电脑,双击将它打开。2、打开之后,在里面找到吸C盘,双击将它打开。3、然后在C盘里面找到Programfiles这个文件,将此文件打开。4、打开之后,在里...

主板bios没有csm选项(主板没有csm怎么办)

对普通用户最大的区别是,符合标准的bootloader必须为UEFI保证二进制兼容。结果:32位UEFI固件只能启动32位操作系统。64位UEFI固件只能启动64位操作系统。由于历史因素、OEM政策,...

电脑中病毒了怎么解决(电脑中病毒了怎么解决方法)

电脑因为中毒而不能上网,解决办法如下:1、首先第一步就是杀毒,对一个菜鸟而言只能用杀毒软件来杀毒了,当然,因为病毒更新往往快于杀毒软件的更新速度,所以杀毒有时候是解决不了问题的额,也就是说很多毒是杀...

win10连接网络打印机(win10连接网络打印机成功后不打印)
  • win10连接网络打印机(win10连接网络打印机成功后不打印)
  • win10连接网络打印机(win10连接网络打印机成功后不打印)
  • win10连接网络打印机(win10连接网络打印机成功后不打印)
  • win10连接网络打印机(win10连接网络打印机成功后不打印)
电脑主机启动一下又灭一直反复

故障分析:电脑开机转一下就停然后再转这是由于电脑硬件接触不良,主板或后续硬件短路,电源故障等造成的。1、内存松了或者金手指氧化这种可能性最大,很多用户都是这种方法解决的。首先我们将内存取下,取下也十分...

如何设开机密码 电脑(如何设置开机电脑密码)

方法如下1.建立开机密码。进入BIOS系统界面,点击键盘的Del按键,点击选项中的设置用户密码。设置完毕进入高级设置,点击密码选项列表的系统密码,点击保存并推出即可;2.设置系统密码。进入系统界...

lenovo联想官方商城(联想电脑官方网站商城)

www.lenovo.com.cn联想集团(下称联想)是一家成立于中国、业务遍及180个市场的全球化科技公司。联想聚焦全球化发展,树立了行业领先的多元企业文化和运营模式典范,服务全球超过10亿用户。作...

我的世界国际版官网(我的世界国际版如何下载)

官网入口:https://www.lezhin.com/这是一款很多用户都在使用的观看各种漫画内容的软件,用户在使用的过程中能够快速了解到各种自己感兴趣的漫画。1可以通过搜索引擎找到植物大战僵尸2国...

我要qq号码免费申请(qq号申请免费申请号码官网)

你可以到腾讯官方网站,去免费注册一个QQ,可以不交费的,是有免费的账号和密码。操作方法:1.打开手机qq,用手指轻点qq图标即可。进入qq后,会进入登录或注册新用户页面。点击新用户,进入注册页面2....

新冠疫苗接种查询(新冠疫苗接种查询不到了)

银川市疫苗接种信息在“我的宁夏”app里查询。打开“我的宁夏”app,打开“健康码”选项,然后打开“疫苗接种查询”,就可以查询到你的疫苗接种信息了。如果你接种了吗两次疫苗,你的健康码就会镶嵌了金边,只...

cdr格式文件怎么打开(cdr格式用什么可以打开)

cdr是什么文件?cdr文件用什么打开?cdr文件是CorelDraw图像制作软件标准的输出格式,与photoshop(PS)图片设计软件类似都属于图片设计软件,需要打开cdr文件我们就需要先了解C...

微软win7系统下载(微软官网window7下载)

因为你是按照名称排列的,试着修改一下排序方式,方法是:在桌面空白处右键,排序方式,修改日期这是因为大部分电脑用的都是GHOST系统,该系统在安装后都会向注册表写入自己的“私人信息”,如:风林火山GH...