需要的小麦量为400千克。假设原始的小麦量为x千克。根据题目中提到的质量损耗16%，可以得到以下方程：x*(1-0.16)=336。解这个方程，可以得到x=400。