Using a tensor as both input and output

Hello, I'm implementing an engine that takes in a state tensor and iterates on it and returns the updated tensor, I have to allocate two tensors, one for the input and one for the output, how can I reuse the input tensor as the output and update it in place? I tried passing the same pointer to the engine but the results are wrong