tensorflow输入shape为(batch_size, in_channels, height, weight)
pytorch的输入shape为(batch_size, height, weight, in_channels)
这在两种框架之间进行模型转换时将会很有用