What is the python equivalent of at::max_out

C++
1

I have a some code written in C++ which I am rewriting to have in python,
I wanted to check whether the two are compatible:

at::max_out(tensor1, tensor2, tensor3)

and

torch.max(tensor1, tensor2, out = tensor3)

I have found the documentation slightly lacking on the matter

2

Yes, max_out is defined as:

Tensor& max_out(const Tensor& self, const Tensor& other, Tensor& result) {
  return at::maximum_out(result, self, other);
}

so the last input tensor should be the out tensor.

3

awesome, thanks for the help