登录免费注册

首页  思维导图  详情



 



PyTorch核心类

PyTorch核心类

2025-12-31 13:41:11   0  举报





AI智能生成

PyTorch核心类

PyTorch核心类

模板推荐

作者其他创作

大纲/内容

张量操作类

Tensor类

基础属性方法

dtype属性

shape属性

device属性

创建操作方法

zeros方法

ones方法

randn方法

from_numpy方法

数学运算方法

add方法

mul方法

matmul方法

sum方法

Storage类

内存管理方法

data_ptr方法

size方法

copy_方法

cuda类

GPU操作方法

cuda方法

is_cuda属性

pin_memory方法

自动微分类

Function类

前向传播方法

forward方法

反向传播方法

backward方法

apply方法

Variable类

梯度计算属性

grad属性

requires_grad属性

retain_grad方法

神经网络类

nn.Module类

模型构建方法

add_module方法

parameters方法

modules方法

训练相关方法

train方法

eval方法

state_dict方法

nn.Linear类

全连接层方法

reset_parameters方法

forward方法

nn.Conv2d类

卷积层方法

padding_mode属性

stride属性

kernel_size属性

nn.LSTM类

循环网络方法

hidden_size属性

num_layers属性

bidirectional属性

优化器类

optim.SGD类

随机梯度下降方法

step方法

zero_grad方法

param_groups属性

optim.Adam类

自适应优化方法

betas属性

eps属性

weight_decay属性

optim.lr_scheduler类

学习率调度方法

step方法

get_lr方法

state_dict方法

数据工具类

Dataset类

数据接口方法

__getitem__方法

__len__方法

transform方法

DataLoader类

数据加载方法

batch_size属性

shuffle属性

num_workers属性

collate_fn方法

Sampler类

采样策略方法

__iter__方法

__len__方法

shuffle方法

 Collect

Get Started

聚类方法

 Collect

Get Started

类图

 Collect

Get Started

Task类图

 Collect

Get Started

类图













评论

0 条评论

下一页



图形选择

思维导图

主题

补充说明

AI生成





修改AI描述

去编辑

重新生成

提示 

关闭后当前内容将不会保存，是否继续？

取消

确定

Document