pytorch api 文档学习笔记
最近感觉PyTorch这个框架使用起来还是非常吃力,看别人的代码也常常有无法理解的地方,决定还是先自己研究一下PyTorch官方文档的API。在这边开个学习记录的坑…
函数常用入参及通用解释原则
函数常见入参:
input
:Tensor格式requires_grad
:布尔值,aotugrad是否需要记录在该Tensor上的操作size
:一般是衡量尺寸的数据,可以是多个数字或collection格式(如list或tuple等)device
:Tensor所处的设备(cuda或CPU),可以用torch.device或直接使用字符串、数值(torch.device的入参)代替。- 使用torch.device作为入参的示例:
torch.randn((2,3), device=torch.device('cuda:1'))
- 使用字符串直接作为入参的示例:
torch.randn((2,3), device='cuda:1')
- 使用数值直接作为入参的示例:
torch.randn((2,3), device=1)
- 使用torch.device作为入参的示例:
dtype
:取值范围见torch.dtype
函数名前加_
是原地操作