How did you resolve this?
I have a tensor that returns from a module without a batch dimension:
torch::Tensor out_tensor = module.forward({input_tensor}).toTensor();
However its is 3 by 512 by 512 and this fails:
out_tensor = out_tensor.squeeze().detach().permute({1, 2, 0});
How can I add the batch dimension?
Thanks