conv2d 在pytorch ， tf 中数据格式

conv2d 在pytorch ，tf中数据格式

pytorch data type is "NCHW"
but tensorflow data type is "NHWC",
when pytorch to onnx, and onnx to tensorflow how this diff process?

onnx only has NCHW support.
Thus, onnx-tf will export a NCHW based graph if you use TensorflowRep.export_graph method.

Conv2D onnx pytorch tf
data format NCHW NCHW NHWC
posted @   michaelchengjl  阅读(9)  评论(0)  编辑  收藏  举报
相关博文：
· pytorch构建并保存模型(.pth) 转化为 torchscript(.pt), 导出为onnx格式
· pytorch模型转换为onnx，并使用netron可视化
· Pytorch：将图像tensor数据用Opencv显示
· tensorflow 进阶 五 ---TensorFlow conv2d原理及实践
· tensorflow 进阶 五 ---TensorFlow conv2d原理及实践
阅读排行：
· 开源Multi-agent AI智能体框架aevatar.ai，欢迎大家贡献代码
· Manus重磅发布：全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后，我竟然真的恢复了删除的微信聊天记录！
· 没有Manus邀请码？试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
点击右上角即可分享
微信分享提示