In addition to sampling layer