Rate this Page

torch.cpu.current_device#

torch.cpu.current_device()[source]#

Returns current device for cpu. Always ‘cpu’.

N.B. This function only exists to facilitate device-agnostic code

Return type

str