The reason I got [1, 512, 1, 1] was  i gave my input as

x = torch.randn([1,3,224,224])
output1 = feature_extractor(x)

But my real feature shape is  [4, 512, 1, 1].
So my batch size is 4? And one input’s feature is [4:512] ?
Thank you.