Запись пользовательского слоя Python с узнаваемыми параметрами в кафе

Question

Я знаю, что this example должен показать, как добавить обучаемые параметры в Python layer с использованием метода add_blob().

Однако я все еще не могу понять, как это можно использовать для установки размеров blob на основе пользовательских параметров.

Здесь есть лучший пример того, как написать слой Python here. Но здесь слой не содержит обучаемых параметров.

Просьба пояснить, как написать пользовательский уровень Python с обучаемыми параметрами.

Answer 1

При добавлении параметров блобы с помощью add_blob(), вы можете reshape дополнительную каплю, либо в setup() методе (справа, когда вы добавляете его), или в reshape() методы слоя.