\# [1 x primary_unit_size x num_classes x output_unit_size x num_primary_unit] \# == [1 x 8 x 10 x 16 x 1152]