目录
  • 文档 >
  • torch 加速器 >
  • torch.accelerator.current_stream
快捷键

torch.accelerator.current_stream

torch.accelerator.current_stream(device=None, /)[source][source]

返回给定设备的当前流。

参数:

设备（ torch.device ，str，int，可选）- 必须匹配当前加速器设备类型的给定设备。如果未提供，则默认使用 torch.accelerator.current_device_index() 。

返回值:

给定设备的当前选定流。

返回类型:

torch.Stream

© 版权所有 PyTorch 贡献者。

使用 Sphinx 构建，主题由 Read the Docs 提供。

文档

查看 PyTorch 的全面开发者文档

查看文档

教程

深入了解初学者和高级开发者的教程

查看教程

资源

查找开发资源并获得您的疑问解答

查看资源

为了分析流量并优化您的体验，我们在本网站上提供 cookie。通过点击或导航，您同意允许我们使用 cookie。作为本网站的当前维护者，Facebook 的 cookie 政策适用。了解更多信息，包括可用的控制选项：cookie 政策。