Understanding the out of place index_put operation
PyTorch Forums
Understanding the out of place index_put operation
I am trying to understand the api of for the out of place C++ index_put function; i.e., the function: at::Tensor at::Tensor::index_put( const c10::List<::std::optional< at::Tensor > >& indices, const at::Tensor& values, bool accumulate = false ) const For example, here is code that I think should start with a 2 by 2 zero matrix and then set the element with index (0, 1) to 5. But instead it seems to set the entire matr...
0 comments
No comments yet.