UserWarning: An output with one or more elements was resized since it had shape

[ptrblck]

Double post from here with follow up.