-
-
Notifications
You must be signed in to change notification settings - Fork 98
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add multiple dtype support #12
Comments
Next step for this is to make |
Device traits are now generic against dtype. Tensor's have dtype as well, but all tensor_ops hard code |
Unclear how necessary this is. Previously the main use case I was thinking of was for class labels, but I don't think its worth supporting all the different data types in tensors just for that, especially since we can just one hot encode the class labels into f32 tensors anyway (see #19). Now the main use cases I can think of are:
Both of those are pretty low priority though |
Closing this for now since at least there are hooks to add other dtypes. If a new dtype should be added, there should be an issue created for that specific dtype |
This will be another generic parameter of all tensors. Most existing operations will likely require float generic.
Related to #9 since it involves an additional generic parameter
The text was updated successfully, but these errors were encountered: