which for ResNet50 models is a [batch_size, 2048] tensor.
return inputs
```
**Listing 2.** Modification to the head of resnet\_main.py.

```