GlobalPooling.ftl 333 Bytes
Newer Older
1 2 3 4 5 6
<#assign input = element.inputs[0]>
<#if element.poolType == "max">
    		${element.name} = brew.max_pool(model, ${input}, '${element.name}', global_pooling=True)
<#elseif element.poolType == "avg">
    		${element.name} = brew.average_pool(model, ${input}, '${element.name}', global_pooling=True)
</#if>
7
<#include "OutputShape.ftl">