Understanding stack_out

(Thomas Johnson) #1

Can someone explain how exactly the _out functions (stack_out, cat_out, etc) work?

Is it correct that torch::stack_out(my_output_tensor, my_vect) is exactly equivalent to torch::stack(my_vect) except that the output tensor will be stored in my_output_tensor, and my_output_tensor will be returned from the function?