gpt4 book ai didi

python - 如果 Edge TPU 编译器仅支持 3 个暗淡张量,如何实现 conv2d 层?

转载 作者:行者123 更新时间:2023-11-30 09:15:18 26 4
gpt4 key购买 nike

我有一个问题,希望很容易回答。

如果我想实现一个简单的 conv2d 层,我需要一个 4 维权重张量,其中所有维度大小通常都大于 1。

但是 Edge TPU 编译器的要求是:“张量可以是 1 维、2 维或 3 维。如果张量超过 3 个维度,则只有 3 个最里面的维度的大小可以大于 1 ”。(https://coral.withgoogle.com/docs/edgetpu/models-intro/)

现在我的问题是:如果只有尺寸大于一的三维张量可用,如何实现转换层?还是我错了?

提前谢谢

最佳答案

我认为它们的意思是输入张量和各层的输出受到这样的限制。正如您正确指出的那样,即使内核张量不满足这些形状要求,编译器当然也接受 conv2d 层!

关于python - 如果 Edge TPU 编译器仅支持 3 个暗淡张量,如何实现 conv2d 层?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57506743/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com