SwapAxes.ftl 232 Bytes
Newer Older
1
<#assign input = element.inputs[0]>
2
3
4
5
<#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})
6
</#if>