label_dtype: An optional `tf.DType` from: `tf.float32

Input: tf.float64

Output: tf.int32

Prompt: tf.int64`. Defaults to `tf.int64`.