torch.tx.InferDType

export type InferDType<T extends readonly TensorOrNumber[]> = T extends readonly [
  infer First,
  ...infer Rest,
]
  ? First extends Tensor<any, infer D, any>
    ? D
    : InferDType<Rest extends readonly TensorOrNumber[] ? Rest : []>
  : 'float32';

Textends readonly TensorOrNumber[]

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