您当前的位置: 首页 >  pytorch

耐心的小黑

暂无认证

  • 2浏览

    0关注

    323博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

pytorch学习笔记十七:官方API文档小总结

耐心的小黑 发布时间:2020-12-06 21:10:31 ,浏览量:2

一、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࿰

关注
打赏
1640088279
查看更多评论
立即登录/注册

微信扫码登录

0.0367s