Handle allocation of pointer array

Author: jwallwork23

examples/5_Autograd/autograd.f90

@@ -24,8 +24,7 @@ program example
   call torch_tensor_from_array(a, in_data, tensor_layout, torch_kCPU)
 
   ! Extract a Fortran array from a Torch tensor
-  allocate(out_data(2))
-  call torch_tensor_to_array(a, out_data)
+  call torch_tensor_to_array(a, out_data, shape(in_data))
   write (*,*) "a = ", out_data(:)
 
   ! Cleanup