Forgot args and kwargs in reshapelayer get output