前言
最近正在学习pytorch,看了up主小土堆的视频,在视频中up主对于官方文档的阅读能力比较强,记录以下up主的总结,阅读文档确实要轻松不少。
一、步骤
以下步骤都已transforms.Normalize为例
在pycharm中,当遇到一个函数或者类不熟悉时,按住ctrl,然后鼠标左键点击该类和函数。可以查看该类或者函数的用法。
1.首先看__init__函数
表明在创建对象的时候需要传入参数mean,std两个参数,inplace使用默认的即可
看参数的注释,应该传入什么类型的参数
从图中得出,mean和std是两个列表,列表长度为输入tensor的通道数
2.然后创建该类的一个对象
3.调用该对象的方法得到输出,明确输出是什么类型
该方法传入的是tensor数据类型,输入的是归一化之后的tensor数据类型。
总结
我对应python面向对象的语法不是很了解,所以阅读文档很困难,后面应该专门学习一下这方面的知识