无需挤压即可获得正确的Tensorflow形状输出

Question

我正在构建一个具有形状（批处理大小，group_size，feature_size）和形状标签（批处理大小，group_size）的网络，然后将其通过以下网络馈入。

在没有Lambda层压缩的网络输出中，最后一个形状是（batch_size，group_size，1），但这样做自然会导致标签形状和该形状之间的形状不匹配，等级分别为3和2。有什么办法可以解决这个问题，还是这是正确的方法？