torch.tx.InferDevice

export type InferDevice<T extends readonly TensorOrNumber[]> = T extends readonly [
  infer First,
  ...infer Rest,
]
  ? First extends Tensor<any, any, infer Dev>
    ? Dev
    : InferDevice<Rest extends readonly TensorOrNumber[] ? Rest : []>
  : DeviceType;

Textends readonly TensorOrNumber[]

Infer device from a tuple of tensors. Uses the first tensor's device, or DeviceType if all scalars.