fix return type of tfds.load in doc
fix return type of tfds.load in doc

Was referring to `tfds.data.Dataset` instead of `tf.data.Dataset`.
tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets