超级会员免费看
本文总结了PyTorch中数据处理、网络构建、张量操作、损失函数、初始化、优化算法、数据集生成及模型加载等方面的常用API，涵盖了torch、torch.nn、torch.nn.functional等模块的关键函数，对于理解和使用PyTorch进行深度学习实践非常有帮助。
一、torch下的常用API，基本上都是对数据进行处理的函数
①操作数据：
torch.is_tensor,torch.set_default_dtype,torch.get_default_dtype,torch.cat,torch.index_select,torch.reshape,torch.squeeze,torch.t,torch.unsqueeze,torch.transpose,torch.take,torch.where
②定义数据(tensor)：
torch.tensor,torch.empty,torch.empty_like,torch.full,torch.full_like,torch.ones，
torch.ones_like,torch.zeros,torch.zeros_like,torch.range,torch.arange,torch.line(log)space
③数据转换(主要是从ndarray转过来)：
torch.from_numpy，torch.as_tensor()
④随机数：
torch.normal，torch.rand，torch.rand_like，torch.randint，torch.randint_like，torch.randn，torch.randn_like，
⑤数学计算
一般数学计算:torch.abs，torch.add，torch.clamp
订阅专栏 解锁全文
2415
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
