Skip to content
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

Set element in an Tensor #2635

Open
DimitriTimoz opened this issue Nov 21, 2024 · 2 comments
Open

Set element in an Tensor #2635

DimitriTimoz opened this issue Nov 21, 2024 · 2 comments

Comments

@DimitriTimoz
Copy link

DimitriTimoz commented Nov 21, 2024

Maybe it's a dumb question and I just didn't well read the doc but Is it possible to assign a value at a position in a tensor?
Something like that:

        self.adjacency_matrix.set((x, y), 1);
@zachcp
Copy link
Contributor

zachcp commented Nov 30, 2024

I believe Tensors are immutable. You can build them up from vectors or iterators but I think it's difficult and maybe impossible to mutate specific elements.

@EricLBuehler
Copy link
Member

@DimitriTimoz yes, tensors are immutable. The way to change the value of a specific element would be to use the Tensor::slice_assign method, which returns a new (copied) tensor. This requires a tensor to copy as the replacement tensor. If this code is in a hot loop and the device is a GPU, I would recommend that you avoid recreating this replacement tensor to avoid device<>host synchronization.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants