Fixup generated function doc example
This makes the example work properly. It'd be nice to set it up to run with doc-tests eventually. It'd also be nice to avoid the complicated `::Type{NTuple{N}}` annotation and simply grab `N = length(dims)` (if/when that is supported).
cc: @tlycken
[av skip]
