Update dtype argument for keras_core.Input #948
Closed
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
At present,the API keras_core.input documentation states that the argument dtype to be passed as a string such as 'tf.float32' etc. However if we pass
tf.dtypes.DType
such astf.float32
,tf.float64
or atorch.dtype
such astorch.float32
,torch.float64
it will works as well as the internal code calls keras_core.backend.standardize_dtype which can converttf.float32
ortorch.float32
to string such as 'float32'.IMO, its better to mention this is documentation that this API can also handle both
tf.dtypes.DType
andtorch.dtype
so that users are aware of this and they are free to choose any of the above ways to pass value to dtype.Attached gist as reference for the above exercise.