SwapAxes.ftl 285 Bytes
Newer Older
Bernhard Rumpe's avatar
BR-sy    
Bernhard Rumpe committed
1
<#-- (c) https://github.com/MontiCore/monticore -->
2
<#assign input = element.inputs[0]>
3
4
5
6
<#assign dim1 = (element.axes[0] + 1)?c>
<#assign dim2 = (element.axes[1] + 1)?c>
<#if mode == "FORWARD_FUNCTION">
        ${element.name} = F.swapaxes(${input}, dim1=${dim1}, dim2=${dim2})
Bernhard Rumpe's avatar
BR-sy    
Bernhard Rumpe committed
7
</#if>