标签： cuda cudnn
我想沿cudnn中的通道维度连接两个张量。但是，我相信没有这种功能可以实现。你们对如何快速做到有什么建议吗？
我可以做的一件事是创建一个新的张量，并使用for循环相应地复制数据。但是，这将花费太多时间。
This code是我在网上找到的。
例如，两个张量的大小均为[2,4,5,5]。输出张量将为[2,8,5,5]