分类 PyTorch 教程 下的文章

stable_baselines3 是什么？它的基本用法是什么？它包含那些算法？
撰写于：  浏览：1546 次  分类：PyTorch 教程
1、Stable-Baselines3 是什么？Stable-Baselines3 是一个基于 PyTorch 的强化学习库，专注于提供高质量、易于使用的强化学习算法实现。它是 Stable-Baselines 的继任者，后者基于 TensorFlow。Stable-Baselines3 旨在简化强化学习算法的使用，同时保持高性能和灵活性。2、St[...]
torch.chunk的应用场景有那些呢？
撰写于：  浏览：1202 次  分类：PyTorch 教程
torch.chunk 在 PyTorch 中有多种应用场景，尤其是在需要将张量分割成多个部分以便并行处理或分批次处理时。以下是一些常见的应用场景：1、数据分批次处理在深度学习中，数据通常需要分批次（batch）进行处理。如果数据已经加载到一个大张量中，可以使用 torch.chunk 将其分割成多个批次。# 假设有一个大的数据集张量 data =[...]
PyTorch中chunk函数的用法
撰写于：  浏览：1414 次  分类：PyTorch 教程
在 PyTorch 中，chunk 是一个用于将张量分割成多个块的函数。它的用法如下所示。1、chunk函数定义1.1、函数签名torch.chunk(input, chunks, dim=0)1.2、参数说明input (Tensor): 要分割的输入张量。chunks (int): 要分割的块数。dim (int, optional): 沿着哪[...]
Transformer VS CNN/RNN
撰写于：  浏览：1412 次  分类：PyTorch 教程
1、在CV领域，Transformer能不能完全取代CNN？在计算机视觉（CV）领域，Transformer是否能完全取代卷积神经网络（CNN）尚无定论，尽管Transformer在多个任务中表现出色，但仍面临一些挑战。Transformer的优势全局建模能力：Transformer通过自注意力机制捕捉全局信息，适合处理长距离依赖，如图像分类、目标[...]
PyTorch view 和 reshape 的区别是什么？
撰写于：  浏览：1123 次  分类：PyTorch 教程
在 PyTorch 中，view 和 reshape 都用于改变张量的形状，但它们的行为和适用场景有所不同。1、view共享内存：view 返回的新张量与原始张量共享内存。这意味着修改其中一个张量会影响另一个。连续性要求：view 要求张量在内存中是连续的（contiguous）。如果张量不连续，需要先调用 contiguous() 方法。适用场景[...]
PyTorch向量转置之后，为什么得到的新向量不是连续的？
撰写于：  浏览：1255 次  分类：PyTorch 教程
在 PyTorch 中，张量的存储方式是按照内存中的连续块来组织的。当我们对张量进行转置操作（如 tensor.t() 或 tensor.transpose()）时，PyTorch 并不会实际改变数据的存储顺序，而是通过修改张量的元数据（如步长 stride 和形状 shape）来实现转置的效果。这种设计是为了避免不必要的数据复制，从而提高效率。然[...]
PyTorch的向量广播的介绍
撰写于：  浏览：1393 次  分类：PyTorch 教程
在PyTorch中，向量广播（Broadcasting）是一种在不同形状的张量之间执行逐元素操作的机制。广播机制使得PyTorch能够自动扩展较小的张量，使其与较大的张量形状兼容，从而进行逐元素操作，而无需显式复制数据。1、广播规则广播遵循以下规则：从尾部对齐：从张量的最右边维度开始，逐维度比较大小。维度大小相等或为1：如果两个张量在某个维度上的大[...]
PyTorch Softmax函数的实现
撰写于：  浏览：1214 次  分类：PyTorch 教程
1、Softmax函数简介Softmax是一种激活函数，它可以将一个数值向量归一化为一个概率分布向量，且各个维度上的概率值相加之后的和为1。在深度学习领域，Softmax函数可以用来作为神经网络的最后一层，用于多分类问题的输出。另外，Softmax层常常和交叉熵损失函数一起结合使用。2、Softmax函数的计算公式对于向量中的每个元素，Softma[...]
Pytorch Embedding的本质理解：查找表
撰写于：  浏览：1057 次  分类：PyTorch 教程
在Pytorch中torch.nn.Embedding层的作用很简单，它的本质就是一张查找表。如下代码所示：import torch import torch.nn as nn torch.manual_seed(1) embedding = nn.Embedding(5, 3) print(embedding.weight)打印出的结果为：[...]
Positional Encoding位置编码的作用
撰写于：  浏览：1141 次  分类：PyTorch 教程
在一门语言中，词语的位置和顺序对句子意思表达至关重要的。传统RNN模型天然有序，在处理句子时，以序列的模式逐个处理句子中的词语，这使得词语的顺序信息在处理过程中被天然的保存下来，并不需要额外的处理。由于Transformer模型没有RNN（循环神经网络）或CNN（卷积神经网络）结构，句子中的词语都是同时进入网络进行处理，所以没有明确的关于单词在源句[...]
深度开发者们对显卡驱动的困惑
撰写于：  浏览：1062 次  分类：PyTorch 教程
自从PC诞生以来，驱动程序就一直如影随形，与硬件产业一同发展。显卡驱动是连接操作系统与显卡硬件的桥梁，显卡驱动程序使操作系统能够识别和使用显卡的基本显示功能。即使没有安装专门的驱动程序，大多数操作系统也能通过通用驱动支持显卡的基本显示功能，但这通常无法充分发挥显卡的全部性能。因为显卡驱动程序包含了一系列优化算法，可以提升显卡处理图形和视频的能力。对[...]
cuDNN的介绍
撰写于：  浏览：1632 次  分类：PyTorch 教程
1、cuDNN是什么？cuDNN全称NVIDIA CUDA® Deep Neural Network library， 是一个用于深度神经网络的GPU加速库。cuDNN包含了为神经网络中常见的计算任务提供高度优化的实现。包括前向卷积、反向卷积、注意力机制、矩阵乘法（matmul）、池化（pooling）和归一化（normalization）等。cu[...]
torch.log()函数的使用介绍
撰写于：  浏览：1279 次  分类：PyTorch 教程
1、torch.log()函数定义torch.log(input, *, out=None) 其作用是返回一个新的张量，其中包含输入元素的自然对数。计算公式如下所示：$$ y_i = log_e(x_i) $$2、torch.log()函数代码举例：>>> a = torch.rand(5) * 5 >>> a[...]
CUDA_VISIBLE_DEVICES环境变量的使用介绍
撰写于：  浏览：1935 次  分类：PyTorch 教程
CUDA_VISIBLE_DEVICES环境变量的简单介绍CUDA 编程模型中的环境变量在 CUDA 应用程序开发中具有重要作用，如需使用，需要在程序运行前在系统中进行设置。环境变量 CUDA_VISIBLE_DEVICES 用于指定 CUDA 应用程序将在哪些 GPU 设备上运行，通常用于控制程序在多 GPU 系统上的 GPU 使用情况，对于单 [...]
DataParallel怎么将输入数据迁移到GPU？
撰写于：  浏览：1229 次  分类：PyTorch 教程
在PyTorch中，DataParallel 是一个模块，它允许你在多个GPU上分布式地训练模型。为了让输入数据自动移动到GPU，你需要做的是将模型和数据包装在 DataParallel 中。以下是一个简单的例子，展示如何使用 DataParallel：import torch import torch.nn as nn from torch.ut[...]
关注公众号，感悟技术与人生

飞燕网
一个踏实、严谨的网站！

专注于PyTorch、强化学习和大模型技术

    友情链接

飞燕网 - 专注于PyTorch、强化学习和大模型技术  ©  2026   运行时间： 2537天   上线时间：2019年3月19日
京ICP备14057001号