torch.device

Software / App

A PyTorch object representing the device (CPU or GPU) on which a tensor is allocated.

Mentioned in 1 video