improve infering of location

this improves inferring the location of a variable. the initial implementation required, that the spatial dimension is the last one of the array. this implementation improves this behaviour