深度学习(三 平台工具)

0x01 云服务工具 google cloab

0x02 pytorch

torch.dataset:数据集

torch.dataloader:在batch中加载组织数据

tensor:高维度阵列,可和numpy互相转换

tensor.transpose:矩阵转置

tensor.squeeze:删除维度

tensor.unsqueeze:添加维度

tensor.cat:沿维度合并tensor

把tensor放到CPU或者GPU中作运算。

定义loss
定义优化器

发表评论

您的电子邮箱地址不会被公开。 必填项已用 * 标注