Я хочу объединить два тензора вдоль измерения канала в cudnn. Тем не менее, я считаю, что такой функции не достичь. Ребята, у вас есть предложения, как это сделать быстро?
Я мог бы создать новый тензор и скопировать данные, используя цикл for. Однако это будет стоить слишком много времени.
Этот код - это то, что я нашел в Интернете.
Например, размеры двух тензоров равны [2,4,5,5]. Выходной тензор будет [2,8,5,5]