Столкнулся с необходимостью написать конволюционный слой кастомный на Торче. Так-то

Question

Столкнулся с необходимостью написать конволюционный слой кастомный на Торче. Так-то

в инете описано всё подробно по вычислениям и проходом ядром по матрице, но... Если писать циклами, то слой отрабатывает очень медленно и жрёт памяти, как не в себя. У кого-то есть опыт какими операциями можно низкоуровнево сделать свёртку так, чтоб это было хотя-бы на порядок медленнее, чем встроенный conv2d, а не на 3+ порядка?

0

06.09.2023

4 ответов

55 просмотров

Лис Электронщик

Linear'ом делай

0

07.09.2023

Sергей Автор вопроса

Лис Электронщик
Linear'ом делай

Жаль, тут стикеры сдать нельзя) например, как сделать линеаром?

0

07.09.2023

Лис Электронщик

Sергей
Жаль, тут стикеры сдать нельзя) например, как сдел...

Очень легко Варианта есть два — либо входной тензор нарезаешь и склеиваешь так, как надо, либо делаешь так с выходным после линеара — зависит от того, какой тензор шире в каналах, выходной или входной

0

07.09.2023

Evgenii Zheltonozhskii🇮🇱 · Accepted Answer

Evgenii Zheltonozhskii🇮🇱

На куде

0

07.09.2023

Похожие чаты

Столкнулся с необходимостью написать конволюционный слой кастомный на Торче. Так-то

4 ответов

Похожие вопросы