Could you explain your use case a bit and what these dimensions would represent?
The standard ResNet would already output a 2D tensor in the shape [batch_size, nb_classes], but I assume you want to use another additional dimension?