chunked/nltk_chunking/model_doc_nat/chunk_15.txt

The reshaped_hidden_states have a shape of (batch, num_channels, height, width) rather than
(batch_size, height, width, num_channels).