すべてのチャレンジ
中級 0件解決済み 正答率0%

Sinusoidal Positional Encoding

Compute the sine and cosine position vectors Transformers add to token embeddings.

このチャレンジで学べること

  • Compute sine and cosine position vectors
  • Use different frequencies per dimension
  • Understand how Transformers know word order

問題

翻訳を表示

Attention ignores word order, so the original Transformer adds a positional encoding to each token embedding. For position pos and dimension j (both starting at 0) in a model of size d_model:

  • even j: PE[pos][j] = sin(pos / 10000^(j / d_model))
  • odd j: PE[pos][j] = cos(pos / 10000^((j - 1) / d_model))

So dimensions 0 and 1 share one frequency, dimensions 2 and 3 the next, and so on.

Implement positional_encoding(length, d_model).

Input

length: The number of positions (0 to length - 1).

d_model: The embedding size.

Output

Return a length × d_model matrix.

制約

1 ≤ length ≤ 200, 1 ≤ d_model ≤ 128

例

例 1
入力:
length = 2
d_model = 4

期待される出力:
expected = [
    [0.0, 1.0, 0.0, 1.0],
    [0.841471, 0.540302, 0.01, 0.99995]
]
入力:
long long length = 2;
long long d_model = 4;

期待される出力:
std::vector<std::vector<double>> expected = {
    {0.0, 1.0, 0.0, 1.0},
    {0.841471, 0.540302, 0.01, 0.99995}
};
入力:
long length = 2;
long d_model = 4;

期待される出力:
double[][] expected = new double[][] {
    new double[] {0.0, 1.0, 0.0, 1.0},
    new double[] {0.841471, 0.540302, 0.01, 0.99995}
};

解説

Position 0 is sin(0) = 0 and cos(0) = 1 for every pair: [0, 1, 0, 1]. Position 1 uses frequencies 1 and 1 / 10000^(2/4) = 0.01: [sin 1, cos 1, sin 0.01, cos 0.01] ≈ [0.841471, 0.540302, 0.01, 0.99995].

例 2
入力:
length = 3
d_model = 1

期待される出力:
expected = [
    [0.0],
    [0.841471],
    [0.909297]
]
入力:
long long length = 3;
long long d_model = 1;

期待される出力:
std::vector<std::vector<double>> expected = {
    {0.0},
    {0.841471},
    {0.909297}
};
入力:
long length = 3;
long d_model = 1;

期待される出力:
double[][] expected = new double[][] {
    new double[] {0.0},
    new double[] {0.841471},
    new double[] {0.909297}
};

解説

With one dimension only sin(pos) is used: [[0], [0.841471], [0.909297]].

参考資料

この課題はCodeUtility用に調整されています。上記の説明と制約に従ってください。

他の人の解答から学ぶ

コミュニティの解答

まだ公開された解答がありません。この課題を解いて最初の解答を共有しましょう。