Commit 38e0e504 authored by Christian Fuß's avatar Christian Fuß
Browse files

added argmax axis to OneHotLayer

parent f37dd998
Pipeline #146886 failed with stages
in 22 seconds
${element.name} = mx.symbol.one_hot(data=${element.inputs[0]},
indices=F.argmax(x), depth=${element.size}))
\ No newline at end of file
indices=mx.symbol.argmax(data=${element.inputs[0]}, axis=1), depth=${element.size}))
<#include "OutputShape.ftl">
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment