Onehot

Top: -

Par: 153 lines

Problem Statement

LM に、値域が 0 以上 16 未満で長さが $256$ の Int ベクトル $T$ が与えられます。

出力として、$32 \times 256$ 行列 $X_{i,j}$ に対して、j = T[j] であれば $1.0$, そうでないなら $0.0$ で埋められた行列を LM に作成してください。

Onehot mini と同様の問題で、ベクトルの長さが $256$ になっています。(4_L2B:2) の通り $4$ 個の L2B に分割されているので、各 PE では長さ $64$ のベクトルだと考えることができます。

PE あたりのサイズが $16$ 倍になったので、Onehot mini をそのまま $16$ 倍に増やすことで Accept が得られます。

$T$ ($0 \le T[{i}] \le 15$): Int $lm[0:64], (256)/((4_L2B:2, 32:1, 2_W:1); B@[PE,MAB,L1B])
?

testcase.vsm