Avoid class mismatch during assignment to memory-mapped output

[epnev]

@conorheins Thanks for this contribution. I modified the function in a more general way to preserve the class of the input for all the different choices of output type.