PyTorch入门教学——dir()函数和help()函数的应用

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当前是否可用。