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