1、简介
- 已知PyTorch是一个工具包,其中包含许多功能函数。dir()函数和help()函数是学习PyTorch包的重要法宝。
- dir():能让我们知道工具包以及工具包中的分隔区有什么东西。
- help():能让我们知道每个工具是如何使用的,即工具的使用方法。
2、使用
2.1、dir()
- 打开创建好的pytorch项目,在python控制台中探索dir()的使用方法。
- 以torch.cuda.is_available()函数为例。
- 输入:dir(torch),可以看到其中还包含了各种各样的包。
- 继续输入:dir(torch.cuda)
- 继续输入:dir(torch.cuda.is_available())
- 此时输出的内容与之前不同,这些为变量,说明is_available()是一个函数。
2.2、help()
- 输入:help(torch.cuda.is_available),查看is_available()函数的作用。
- 【注】函数的括号要去掉。
- 返回一个布尔值,表明CUDA当前是否可用。