fix flatten issue in converted tensorflowjs model

[ShuangLiu1992]

explicitly set the size to match the input instead of using -1, so that tensorflowjs won't complain about shape mismatch [None, actual_size]