问TensorRT如何处理数据？
EN

Stack Overflow用户

提问于 2019-08-29 17:02:56

回答 1查看 48关注 0票数 0

在Tensorflow中，张量格式是NxWxHxC并通过网络。

当Tensorflow模型转换为TensorRT引擎时，如何处理张量数据。与Same中相同的形状或扁平的。

展平表示，例如，1x3x4x3张量

[[[1,2,3],[4,5,6],[7,8,9],[10,11,12]],
     [[1,2,3],[4,5,6],[7,8,9],[10,11,12]],
     [[1,2,3],[4,5,6],[7,8,9],[10,11,12]]]

被展平为

[1,2,3,4,5,6,7,8,9,10,11,12,1,2,3,4,5,6,7,8,9,10,11,12,1,2,3,4,5,6,7,8,9,10,11,12]

单个数组并已处理。

回答已采纳

发布于 2019-09-04 14:25:57

TensorRT以NCHW格式处理数据。当数据被展平时，它是NCHW格式。这意味着，数据被展平为

1,4,7,10,1,4,7,10,1,4,7,10,2,5,8,11,2,5,8,11,2,5,8,11,3,6,9,12,3,6,9,12,3,6,9,12

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/57706675

复制

相似问题

问TensorRT如何处理数据？EN