《PyTorch面试精华》


1 前言
    1.1  PyTorch安装

    1.2  显卡驱动的困惑

    1.3  CUDA安装注意事项

    1.4  cuDNN的介绍

    1.5  Pytorch Lightning介绍

    1.6  PyTorch学习之道

    1.7  PyTorch快速入门

    1.8  PyTorch调参之道

    1.9  PyTorch调参套件

    1.10  手动创建虚拟环境

2 深度学习之数学基础
    2.1  希腊字母解读

    2.2  梯度的物理意义

    2.3  图解梯度下降法

    2.4  图解梯度上升法

    2.5  自然梯度

    2.6  泰勒公式的介绍

    2.7  信息与信息熵

    2.8  重要性采样

    2.10  欧几里得范数

    2.11  特征值和特征向量

    2.12  似然函数的理解

    2.13  矩阵秩的深刻理解

3 PyTorch入门疑难点
4 PyTorch全局设置
    4.1  全局设置当前设备

    4.2  全局设置浮点精度

5 PyTorch GPU分布式训练
    5.1  PyTorch GPU基础操作

    5.2  DataParallel用法详解

    5.3  GPU分布式训练模型

    5.4  CUDA_VISIBLE_DEVICES

    5.5  device详细说明

    5.6  to(device)和.cuda()

    5.7  CUDA设备索引

    5.8  GPU设备索引

6 向量的基础与核心
    6.1  Tensor的组成与存储

    6.2  Tensor的grad属性

    6.4  Tensor的叠加

    6.5  禁用梯度计算

    6.6  向量的保存和加载

    6.7  参数向量

    6.8  叶子节点

    6.9  detach原理

    6.10  requires_grad属性

    6.11  Tensor与Numpy互换

    6.12  张量cat操作

    6.13  零维张量

    6.15  squeeze/unsqu...函数

    6.16  argmax和max的区别

    6.17  torch.as_tensor的应用

7 神经网络基础
    7.2  PyTorch计算图

    7.3  查看网络权重参数

    7.4  保存模型

    7.5  Adam相关面试题

    7.6  Train模式和Eval模式

    7.7  线性网络

    7.8  双线性网络

    7.9  惰性线性层

    7.10  PyTorch中的自动微分

    7.12  Dropout机制

    7.13  半精度训练

    7.14  Xavier初始化

    7.15  注意力机制

    7.16  Dataset数据处理

    7.17  StepLR学习率调度器

    7.18  词嵌入的理解

    7.19  TensorDataset的使用

    7.20  模型的保存与加载

    7.21  ModuleList和Sequential

    7.22  Batch Normalization介绍

8 计算机视觉基础知识
    8.1  通道的深刻理解

    8.2  1x1卷积的作用

    8.3  特征提取和可视化

    8.4  反卷积的推导

    8.5  理解卷积

    8.7  空洞卷积

    8.8  池化层的作用

    8.9  感受野与特征图

    8.10  NMS算法

    8.11  特征图尺寸计算

9 循环神经网络基础
    9.2  RNN的介绍

10 注意力机制
    10.1  位置编码的作用

    10.2  位置编码的种类

    10.4  Embedding本质理解

    10.6  Transformer VS CNN/RNN

    10.7  ELMo介绍

11 PyTorch归一化
    11.2  层归一化技术详解

12 激活函数相关内容
    12.1  激活函数简介

    12.2  万能逼近定理

    12.3  指数函数的学习

    12.4  Sigmoid函数的介绍

    12.5  Tanh函数的介绍

    12.6  Softmax函数的实现

    12.7  ReLU函数的介绍

    12.8  Leaky Relu函数的介绍

    12.9  ReLu与非线性的理解

    12.10  Parametric ReLU函数

    12.11  ELU函数介绍

    12.12  神经元死亡的问题

13 思考题的答案
    13.1  思考题的答案解密

detach原理

创建时间:2024-09-10 更新时间:2024-09-15 阅读次数:1712 次

飞燕提醒:detach()函数的一大作用是将将张量从GPU移动到CPU时,在本教程的《tensor与numpy互换小节》中大家可以深刻体会。

1、detach函数的作用

在PyTorch中,detach()函数的作用有两个:

(1)将张量从当前计算图中分离出来,从而不需要跟踪张量的梯度变化

(2)将张量从GPU移动到CPU时。

代码例子1:

import torch

x = torch.tensor(2.0, requires_grad = True)

print("Tensor:", x)

x_detach = x.detach()
print("Tensor with detach:", x_detach)

结果为:

Tensor: tensor(2., requires_grad=True)
Tensor with detach: tensor(2.)

代码分析:在上述输出中,分离后的张量没有requires_grad=True

代码例子2:

import torch

x = torch.rand(3, requires_grad = True)
print("x:", x)

y = 3 + x
z = 3 * x.detach()

print("y:", y)
print("z:", z)

输出结果为:

x: tensor([0.5656, 0.8402, 0.6661], requires_grad=True)
y: tensor([3.5656, 3.8402, 3.6661], grad_fn=<AddBackward0>)
z: tensor([1.6968, 2.5207, 1.9984])

2、虽然共享内存,但又相互独立

在PyTorch中,detach()函数是一个非常有用的方法,它主要用于创建一个与原张量tensor逻辑上分离的新张量。尽管它们共享相同的存储空间,但任何对detach()返回的新张量的修改都不会影响到原始张量,反之亦然。如下代码所示:

import torch

x = torch.tensor([1.0])
y = x + 1.0
print("y值:", y)

y_ = y.detach()
y_ = y_ + 1.0
print("分离出的y_值:", y_)

print("y值:", y)

结果为:

y值: tensor([2.])
分离出的y_值: tensor([3.])
y值: tensor([2.])

3、detach的应用场景

  • 控制梯度更新:当你想控制计算图特定部分的梯度更新时,detach很有用。例如,在强化学习中,你可能希望将动作值从策略网络中分离出来,以防止梯度流过它们。

  • 创建独立张量:如果你需要创建一个独立于当前计算图的张量,detach会有所帮助。当你想在不影响梯度计算的情况下存储中间结果时,detach可能很有用。

  • 调试和可视化:在调试和可视化过程中,分离张量可能会有所帮助。通过分离张量,你可以确保调试或可视化代码不会干扰梯度计算。

本教程共117节,当前为第49节!
本教程最新修订时间为:2026-05-08 11:10:53