fix: torch backend mean reverted to work with keepdim, out args, as well as dtype

fix: torch backend mean reverted to work with keepdim, out args, as well as dtype #41722