fix squeeze

[simonJJJ]

No description provided.

[github-actions]

Speed stats:

GPU Name: GeForce GTX 1080 

PyTorch resnet50 time: 140.0ms (= 7001.9ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 128.2ms (= 6409.9ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
Relative speed: 1.09 (= 140.0ms / 128.2ms)

PyTorch resnet50 time: 84.5ms (= 4222.9ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 74.4ms (= 3719.3ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
Relative speed: 1.14 (= 84.5ms / 74.4ms)

PyTorch resnet50 time: 55.9ms (= 2796.9ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 49.4ms (= 2471.5ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
Relative speed: 1.13 (= 55.9ms / 49.4ms)

PyTorch resnet50 time: 49.7ms (= 2483.8ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 41.1ms (= 2056.3ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
Relative speed: 1.21 (= 49.7ms / 41.1ms)

PyTorch resnet50 time: 44.5ms (= 2223.5ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 40.3ms (= 2013.3ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
Relative speed: 1.10 (= 44.5ms / 40.3ms)