ExpandDims.ftl 206 Bytes
Newer Older
Bernhard Rumpe's avatar
BR-sy    
Bernhard Rumpe committed
1
<#-- (c) https://github.com/MontiCore/monticore -->
2
3
4
<#assign axis = (element.axis + 1)?c>
<#if mode == "FORWARD_FUNCTION">
        ${element.name} = F.expand_dims(${element.inputs[0]}, axis=${axis})
Bernhard Rumpe's avatar
BR-sy    
Bernhard Rumpe committed
5
</#if>