device

Tensor.device()

Retrieves the device type on which the tensor data is stored.

This method returns if the tensor data is stored on host CPU or Houmo device.

Returns:

device -- The device type on which the tensor data is stored.

Return type:

Device