快捷方式

batch_to_device

torchtune.utils.batch_to_device(batch: dict, device: device) None[source]

此函数接受张量的字典(或嵌套字典),并将它们全部设置为相同的设备。此实用程序旨在用于将数据批次移动到设备,更新是就地进行的。

参数:
  • batch (dict) – 张量的字典或更深层嵌套的张量字典。

  • device (torch.device) – 要将张量移动到的 torch 设备

引发:

AttributeError – 如果批处理字典包含张量以外的任何内容

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源