Python的TensorFlow的NHWC和NVHW数据格式举例

TensorFlow中的NHWC和NVHW是两种不同的数据格式，NHWC表示数据的通道维度在最后一维，而NVHW表示数据的通道维度在第一维。

下面是NHWC和NVHW数据格式的示例：

NHWC格式示例：

假设我们有一个形状为[batch_size, height, width, channels]的4D张量，其中batch_size表示批次大小，height表示高度，width表示宽度，channels表示通道数。在NHWC格式下，数据的通道维度在最后一维，例如一个形状为[1, 28, 28, 3]的张量表示一个批次大小为1，高度为28，宽度为28，通道数为3的图像。

NVHW格式示例：

假设我们有一个形状为[batch_size, channels, height, width]的4D张量，在NVHW格式下，数据的通道维度在第一维。例如一个形状为[1, 3, 28, 28]的张量表示一个批次大小为1，通道数为3，高度为28，宽度为28的图像。