Task: Convert between NHWC and NCHW formats in TensorFlow
Task: Convert between NHWC and NCHW formats in TensorFlow: a task in Terminal-Lego-15k (Harbor dataset). Write a Python script that demonstrates how to convert tensors between NHWC (batch, height, width, channels) and NCHW (batch, channels, height, width) formats in TensorFlow.
The task
Write a Python script that demonstrates how to convert tensors between NHWC (batch, height, width, channels) and NCHW (batch, channels, height, width) formats in TensorFlow.
Part of PrimeIntellect/Terminal-Lego-15k.