Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Implement get_dtype and get_device_type for tensors #248

Closed
jwallwork23 opened this issue Jan 21, 2025 · 0 comments · Fixed by #251
Closed

Implement get_dtype and get_device_type for tensors #248

jwallwork23 opened this issue Jan 21, 2025 · 0 comments · Fixed by #251
Assignees
Labels
autograd Tasks towards the online training / automatic differentiation feature enhancement New feature or request

Comments

@jwallwork23
Copy link
Contributor

These, in combination with the existing get_rank, get_shape, and get_device_index, will allow us to create one tensor from another using the same setup.

This will be useful for autograd because the default is to compute the gradient w.r.t. a tensor of ones of the appropriate shape.

We should also make get_device_index a method of torch_tensor, as with get_rank and get_shape.

@jwallwork23 jwallwork23 added autograd Tasks towards the online training / automatic differentiation feature enhancement New feature or request labels Jan 21, 2025
@jwallwork23 jwallwork23 self-assigned this Jan 21, 2025
@jwallwork23 jwallwork23 changed the title Implement get_dtype and get_device for tensors Implement get_dtype and get_device_type for tensors Jan 24, 2025
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
autograd Tasks towards the online training / automatic differentiation feature enhancement New feature or request
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant