fixed stacked AE cost , finally get diff < 1e-9