首页
学习
活动
专区
圈层
工具
发布

#pytorch

PyTorch是一个基于Torch的 开源机器学习库,用于自然语言处理等应用程序。

NVIDIA ALCHEMI工具包:加速AI驱动化学与材料科学模拟

用户11764306

机器学习原子间势能(MLIP)正在变革计算化学和材料科学领域。MLIP使得原子模拟能够将计算成本高昂的量子化学的保真度与AI的扩展能力相结合。

20400

【Java深度学习】PyTorch On Java 系列课程 第七章 14 :常用模型结构CNN RNN Pool[PyTorch Java 硕士研一课程]

用户11467648

池化层 (nn.MaxPool2d):此层减小特征图的空间尺寸(高和宽),使表示更紧凑,并对特征位置的变化略微更具鲁棒性。

7710

【Java深度学习】PyTorch On Java 认识Javacpp-PyTorch 的强大[PyTorch Java ]

用户11467648

11010

【Java深度学习】PyTorch On Java 系列课程 第六章 13 :模型训练高阶【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

调用optimizer.step()会遍历在优化器初始化时注册的所有参数。对于每个参数p,它会使用存储在p.grad中的梯度来更新参数值p.data。

11310

【Java深度学习】PyTorch On Java 系列课程 第六章 12 :模型训练【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

前向传播: 将输入特征批次送入你的模型。模型通过其层处理数据,应用学习到的权重和激活函数,最终生成一批预测或输出。

8710

【Java深度学习】PyTorch On Java 系列课程 第五章 11 :数据集高级 【AI Infra 3.0】PyTorch Java 硕士研一课程

用户11467648

运行此代码将展示 DataLoader 如何产生数据批次。注意每个批次打印的形状反映了 batch_size(除了可能的最后一个批次)。

10610

【Java深度学习】PyTorch On Java 系列课程 第五章 10 :数据集【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

__getitem__ 接受一个索引 idx,获取对应的特征和标签,将它们转换为 PyTorch 张量,并以元组形式返回。这种转换为张量的操作在 __getit...

9210

【Java深度学习】PyTorch On Java 系列课程 第四章 09 :神经网络进阶【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

这创建了一个模型,其中输入数据首先经过nn.Linear(784, 128),然后应用nn.ReLU()激活,最后结果通过nn.Linear(128, 10)。...

8810

【Java深度学习】PyTorch On Java 系列课程 第四章 08 :神经网络【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

nn.Module的一个重要特点是其自动注册和管理可学习参数的能力。当你在__init__方法中将一个PyTorch层(如nn.Linear、nn.Conv2d...

9410

【Java深度学习】PyTorch On Java 系列课程 第三章 07 :张量梯度【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

我们定义了 x、w 和 b,并设置 requires_grad=True,将它们标记为我们想要梯度的叶节点。

7510

【Java深度学习】PyTorch On Java 系列课程 第三章 06 :张量自动微分【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

x (数据=2.0, requires_grad=True)w (数据=3.0, requires_grad=True)b (数据=1.0, requires_...

9510

【Java深度学习】PyTorch On Java 系列课程 第二章 05 :张量广播机制【Ai Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

下一个维度:a 为 2,b 在此处没有维度(隐式大小为 1)。兼容。结果维度大小为 2。

7110

【Java深度学习】PyTorch On Java 系列课程 第二章 04 :张量高级操作【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

注意,访问单个元素会返回一个包含单个值的torch.Tensor(一个0维张量或标量),而不是一个标准的Python数字,除非您使用.item()明确提取它。元...

9910

【Java深度学习】PyTorch On Java 系列课程 第一章 03 :张量基本操作【Ai Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

PyTorch 也提供许多操作的就地版本。这些操作直接修改张量,而不创建新对象,这可以节省内存。就地函数通常可以通过名称中末尾的下划线 _ 来识别(例如,add...

8710

【Java深度学习】PyTorch On Java 系列课程 第一章 02 :张量操作【AI Infra 3.0】[PyTorch Java 硕士研一课程]

用户11467648

请注意 PyTorch 如何根据输入整数判断数据类型为 torch.int64(一个 64 位整数)。

9410

【Java深度学习】PyTorch On Java 系列课程 第一章 01 :入门与环境搭建 【AI Infra 3.0】PyTorch Java 硕士研一课程

用户11467648

现在,访问 PyTorch 网站,选择您的操作系统、Conda、Python 版本和所需的 CUDA 版本(或 CPU)。复制生成的命令。它看起来会像这样(仅为...

12210

Pytorch On Java 你的第一个java版本的【真】 神经网络 [AI Infra 3.0]

用户11467648

如何用Java 写 全真的神经网络 一直是java 众多程序员的梦想,奈何你们寻不到真经,被各种玩具框架 妖魔鬼怪 挟持认知,一个个都以为必须用jni 调pyt...

7610

PyTorch On Java 体系 视频课程已发布 AI Infra3.0 【硕士研一课程】

用户11467648

可以这么说,cpp libtorch 作为人工智能的工业母鸡 ,pytorch python只不过是一层 基于torch.C 的一层皮 一层wrapper 而已...

9310
领券