No con la interfaz de operaciones actual. Tendría que escribir una nueva operación que tome el tamaño del kernel como un tensor de entrada. La implementación del kernel se puede refactorizar para utilizar la mayor parte de la implementación actual de max_pool.
Clausura. No dude en volver a abrir si esta es una solicitud de función que vale la pena rastrear.
+1
Me gustaría solicitar esa función.
Comentario más útil
+1
Me gustaría solicitar esa función.