batch,height,widith,channel
batch为一批数据的图片数目，height为行，w为列，c为三个颜色层，四个轴axis
0，1，2，3
tf：
b,h,w,c
pytorch：
b,c,h,w
互通时需要对content进行转维度，比如out=transpose（input,[0,3,1,2])，就是tf转成pytorch的数据
tf和pytroch数据维度不同
最新推荐文章于 2025-06-25 20:21:37 发布
本文详细解析了深度学习框架中数据维度的概念，包括batch、height、width和channel的含义，以及TensorFlow与PyTorch在数据组织上的差异。通过具体示例说明了如何在两种框架间进行数据维度的转换。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
扫一扫
325
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
