Não com a interface operacional atual. Você teria que escrever um novo op que leva o tamanho do kernel como um tensor de entrada. A implementação do kernel pode ser refatorada para usar a maior parte da implementação max_pool atual.
Fechando. Sinta-se à vontade para reabrir se esta for uma solicitação de recurso que vale a pena rastrear.
+1
Eu gostaria de solicitar esse recurso.
Comentários muito úteis
+1
Eu gostaria de solicitar esse recurso.