 Input size is [N, 1, 32, 1025]. Output size is [N, 1]. See full model below. 