言語モデルは未来の単語を盗み見てはいけません。因果的アテンションは未来をマスクします。クエリ行 i が注目できるのはキー行 0 から i までだけです。
各行 i について(queries、keys、values はいずれも n 行):
- 許可された各キー j ≤ i のスコアを計算します:score = (queries[i] · keys[j]) / √d。d は keys の列数です。
- 許可されたスコアだけでソフトマックスを計算します(e^x を計算する前に最大値を引きます)。
- 出力行 i = j ≤ i にわたる Σ weight_j × values[j]。
causal_attention(queries, keys, values) を実装してください。
入力
queries, keys: n 行、各行 d 個の数。
values: n 行、各行 v 個の数。
出力
n × v の出力行列を返してください。
制約
1 ≤ n ≤ 50、1 ≤ d, v ≤ 32
-10 ≤ 各値 ≤ 10
例
入力:
queries = [
[0.0],
[0.0],
[0.0]
]
keys = [
[1.0],
[2.0],
[3.0]
]
values = [
[3.0],
[6.0],
[9.0]
]
期待される出力:
[
[3.0],
[4.5],
[6.0]
]
解説
スコアはすべて 0 なので、各行は見られる値の平均になります。行 0 は 3 だけ、行 1 は 3 と 6 の平均(4.5)、行 2 は3つすべての平均(6)です。
入力:
queries = [
[1.0, 0.0],
[1.0, 0.0]
]
keys = [
[1.0, 0.0],
[1.0, 0.0]
]
values = [
[1.0, 2.0],
[3.0, 4.0]
]
期待される出力:
[
[1.0, 2.0],
[2.0, 3.0]
]
解説
行 0 は自分だけを見られるので values[0] = [1, 2] をそのままコピーします。行 1 は2つのキーに同じスコアを与えるので、その平均 [2, 3] になります。
参考資料
-
例 1 未実行入力:
queries = [ [0.0], [0.0], [0.0] ] keys = [ [1.0], [2.0], [3.0] ] values = [ [3.0], [6.0], [9.0] ]期待される出力:[ [3.0], [4.5], [6.0] ]あなたの出力 -
例 2 未実行入力:
queries = [ [1.0, 0.0], [1.0, 0.0] ] keys = [ [1.0, 0.0], [1.0, 0.0] ] values = [ [1.0, 2.0], [3.0, 4.0] ]期待される出力:[ [1.0, 2.0], [2.0, 3.0] ]あなたの出力
提出時に非公開テスト4件も実行されます。 失敗したテストがあれば、そのテストが表示されます。チャレンジを解くとすべて見られます。
Online Java Compiler
Write, compile, run, and debug Java code online in a full-featured browser IDE. Work with multiple files, save your workspace, and share or collaborate in real time - no setup required.
☕ このJavaオンライン実行ツールについて
CodeUtility Java Executorは、ブラウザ上で直接Javaプログラムを記述・実行できるツールです - インストールやJDKの設定、IDEは不要です。安全なサンドボックス上で動作し、実際のJavaランタイム バージョン8、11、17、21、そして最新に対応しています。
このツールは、本物のjavacコンパイラとJavaランタイム環境をクラウドで使用してコードをコンパイル・実行します。
そのため、クラスやメソッド、ロジックをローカルとまったく同じ挙動でテストできます。
オブジェクト指向の練習、面接対策、統合前のスニペット検証など、あらゆる用途で CodeUtility Java Executorは高速かつ信頼できる実行環境を提供し、どんなデバイスからでもすぐに本物のJavaコードを実行できます。
同じ実行環境で、複数ファイル対応エディター、分離された実行ターミナル、ランタイム別ライブラリ、永続ワークスペース、共有、リアルタイム共同作業を利用できます。
この実行環境の使い方
ランタイムを選択し、ファイルを開くか作成してコードを記述し、[実行]を選択します。出力はターミナルに表示されます。サインインしたユーザーは、ワークスペースにファイルを保存し、ライブラリをインストールして、共有を通じて共同作業できます。
- バージョンを選択して現在のファイルを編集するか、ワークスペース内の別のファイルを開きます。
- コードを実行し、求められた場合はターミナルに入力して、保存された実行履歴を確認します。
- 複数のコードスニペットが必要な作業では、ライブラリ、ワークスペースツール、共有、チャットを利用します。
💡 上のエディタで試せるJavaの基本とサンプル
1. 変数と定数の宣言
Javaでは各変数の型を宣言する必要があります。定数にはfinalを使います。
int age = 30;
double pi = 3.14159;
char grade = 'A';
String name = "Alice";
boolean isActive = true;
// 定数
final int MAX_USERS = 100;
final String COMPANY = "CodeUtility";
2. 条件分岐(if / switch)
制御フローにはif、else if、switchを使います。
int x = 2;
if (x == 1) {
System.out.println("1");
} else if (x == 2) {
System.out.println("2");
} else {
System.out.println("その他");
}
switch (x) {
case 1:
System.out.println("1");
break;
case 2:
System.out.println("2");
break;
default:
System.out.println("その他");
}
3. ループ
繰り返しにはfor、while、do-whileを使います。
for (int i = 0; i < 3; i++) {
System.out.println(i);
}
int n = 3;
while (n > 0) {
System.out.println(n);
n--;
}
4. 配列
配列は同じ型の要素を固定長で保持します。
int[] numbers = {10, 20, 30};
System.out.println(numbers[1]);
5. ArrayListの操作
ArrayListは可変長のリストに使います。
import java.util.ArrayList;
ArrayList<Integer> nums = new ArrayList<>();
nums.add(1);
nums.add(2);
nums.add(3);
nums.remove(Integer.valueOf(2));
for (int num : nums) {
System.out.print(num + " ");
}
6. コンソール入出力
入力にはScanner、出力にはSystem.outを使います。
import java.util.Scanner;
Scanner scanner = new Scanner(System.in);
System.out.print("お名前を入力してください: ");
String name = scanner.nextLine();
System.out.println("こんにちは、" + name);
7. メソッド
戻り値の型と引数を指定してメソッドを定義します。
public static int add(int a, int b) {
return a + b;
}
System.out.println(add(3, 4));
8. HashMap
HashMapはキーと値のペアを格納するために使います。
import java.util.HashMap;
HashMap<String, Integer> ages = new HashMap<>();
ages.put("Alice", 30);
System.out.println(ages.get("Alice"));
9. 例外処理
実行時例外はtryとcatchで処理します。
try {
throw new Exception("問題が発生しました");
} catch (Exception e) {
System.out.println(e.getMessage());
}
10. ファイル入出力
ファイル操作にはjava.nio.fileのFilesとPathsを使います。
import java.nio.file.*;
import java.io.IOException;
Files.writeString(Paths.get("file.txt"), "こんにちは、ファイル");
String content = Files.readString(Paths.get("file.txt"));
System.out.println(content);
11. 文字列操作
Javaの文字列はlength()、substring()、contains()など多くのメソッドをサポートします。
String text = "Hello World";
System.out.println(text.length());
System.out.println(text.substring(0, 5));
System.out.println(text.contains("World"));
12. クラスとオブジェクト
Javaはクラスとインスタンスを用いたオブジェクト指向プログラミングをサポートします。
class Person {
String name;
Person(String name) {
this.name = name;
}
void greet() {
System.out.println("こんにちは、私は " + name);
}
}
Person p = new Person("Alice");
p.greet();