torch

Tensors and Neural Networks with 'GPU' Acceleration

Provides functionality to define and train neural networks similar to 'PyTorch' by Paszke et al (2019) <doi:10.48550/arXiv.1912.01703> but written entirely in R using the 'libtorch' library. Also supports low-level tensor operations and 'GPU' acceleration.

0.17.0 MIT + file LICENSE 1 个版本
主页
Bug 反馈
安装
install.packages("torch")
remotes::install_cran("torch")
版本列表
0.17.0 2026-04-11
相关教程
AI 性能分析工具：PyTorch Profiler 与 nsys
学习使用 PyTorch Profiler、NVIDIA Nsight 等工具剖析模型计算图，定位算子性能瓶颈和内存使用。
深度学习 PyTorch：张量、自动求导与模块
使用 PyTorch 搭建神经网络，掌握 Tensor 运算、Autograd 和 nn.Module。
FSDP：PyTorch 原生的全分片数据并行
学习 PyTorch FSDP 的原理与用法，在数据并行中分片模型参数、梯度和优化器状态，并支持多种分片策略以平衡内存与通信。
PyTorch 自然语言处理：Transformers 与微调
基于 PyTorch 和 Hugging Face 实现文本分类、命名实体识别、机器翻译等 NLP 任务，掌握模型微调与推理优化。