NHWC与NCHW

65 阅读 0 评论 43 点赞

我是靠谱客的博主喜悦苗条，最近开发中收集的这篇文章主要介绍NHWC与NCHW，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

http://blog.csdn.net/jningwei/article/details/78156011

区别

NHWC

[batch, in_height, in_width, in_channels]

NCHW

[batch, in_channels, in_height, in_width]

转换

NHWC –> NCHW：

import tensorflow as tf

x = tf.reshape(tf.range(24), [1, 3, 4, 2])
out = tf.transpose(x, [0, 3, 1, 2])

print x.shape
print out.shape
 
 1
2
3
4
5
6
7

(1, 3, 4, 2)
(1, 2, 3, 4)
 
 1
2

NCHW –> NHWC：

import tensorflow as tf

x = tf.reshape(tf.range(24), [1, 2, 3, 4])
out = tf.transpose(x, [0, 2, 3, 1])

print x.shape
print out.shape
 
 1
2
3
4
5
6
7

(1, 2, 3, 4)
(1, 3, 4, 2)

最后

以上就是喜悦苗条为你收集整理的NHWC与NCHW的全部内容，希望文章能够帮你解决NHWC与NCHW所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。

本文分类：TensorFlow
浏览次数：65 次浏览
发布日期：2024-06-28 22:40:02
本文链接：https://www.kaopuke.com/article/k-p-k_13_u_7_o_22_f3_14_z_2_4.html

NHWC与NCHW

概述

http://blog.csdn.net/jningwei/article/details/78156011

区别

NHWC

NCHW

转换

NHWC –> NCHW：

NCHW –> NHWC：

最后

评论列表共有 0 条评论

发表评论取消回复

NHWC与NCHW

概述

http://blog.csdn.net/jningwei/article/details/78156011

区别

NHWC

NCHW

转换

NHWC –> NCHW：

NCHW –> NHWC：

最后

相关文章

评论列表共有 0 条评论

发表评论 取消回复

发表评论取消回复