โจทย์
เลเยอร์ของโครงข่ายประสาทเทียม attention และการเรียนรู้ของเครื่องส่วนใหญ่ ล้วนลงเอยที่ การคูณเมทริกซ์ สำหรับเมทริกซ์ A ขนาด n × k และเมทริกซ์ B ขนาด k × m ผลคูณ C = A × B มีขนาด n × m โดยแต่ละสมาชิกคือผลคูณจุดของแถวหนึ่งของ A กับคอลัมน์หนึ่งของ B:
C[i][j] = A[i][0] × B[0][j] + A[i][1] × B[1][j] + … + A[i][k-1] × B[k-1][j]
จงเขียน matmul(a, b)
อินพุต
a: เมทริกซ์ n × k
b: เมทริกซ์ k × m (จำนวนแถวเท่ากับจำนวนคอลัมน์ของ a)
เอาต์พุต
คืนค่าเมทริกซ์ผลคูณขนาด n × m
เงื่อนไข
1 ≤ n, k, m ≤ 50
-100 ≤ แต่ละค่า ≤ 100
ตัวอย่าง
ตัวอย่างที่ 1
อินพุต:
a = [
[1.0, 2.0],
[3.0, 4.0]
]
b = [
[5.0, 6.0],
[7.0, 8.0]
]
ผลลัพธ์ที่คาดหวัง:
expected = [
[19.0, 22.0],
[43.0, 50.0]
]อินพุต:
std::vector<std::vector<double>> a = {
{1.0, 2.0},
{3.0, 4.0}
};
std::vector<std::vector<double>> b = {
{5.0, 6.0},
{7.0, 8.0}
};
ผลลัพธ์ที่คาดหวัง:
std::vector<std::vector<double>> expected = {
{19.0, 22.0},
{43.0, 50.0}
};อินพุต:
double[][] a = new double[][] {
new double[] {1.0, 2.0},
new double[] {3.0, 4.0}
};
double[][] b = new double[][] {
new double[] {5.0, 6.0},
new double[] {7.0, 8.0}
};
ผลลัพธ์ที่คาดหวัง:
double[][] expected = new double[][] {
new double[] {19.0, 22.0},
new double[] {43.0, 50.0}
};คำอธิบาย
C[0][0] = 1×5 + 2×7 = 19, C[0][1] = 1×6 + 2×8 = 22, C[1][0] = 3×5 + 4×7 = 43 และ C[1][1] = 3×6 + 4×8 = 50
ตัวอย่างที่ 2
อินพุต:
a = [
[1.0, 0.0, 2.0]
]
b = [
[3.0],
[4.0],
[5.0]
]
ผลลัพธ์ที่คาดหวัง:
expected = [
[13.0]
]อินพุต:
std::vector<std::vector<double>> a = {
{1.0, 0.0, 2.0}
};
std::vector<std::vector<double>> b = {
{3.0},
{4.0},
{5.0}
};
ผลลัพธ์ที่คาดหวัง:
std::vector<std::vector<double>> expected = {
{13.0}
};อินพุต:
double[][] a = new double[][] {
new double[] {1.0, 0.0, 2.0}
};
double[][] b = new double[][] {
new double[] {3.0},
new double[] {4.0},
new double[] {5.0}
};
ผลลัพธ์ที่คาดหวัง:
double[][] expected = new double[][] {
new double[] {13.0}
};คำอธิบาย
เมทริกซ์ 1 × 3 คูณเมทริกซ์ 3 × 1 ได้เมทริกซ์ 1 × 1: 1×3 + 0×4 + 2×5 = 13
แหล่งอ้างอิง
โจทย์นี้ปรับให้เหมาะกับ CodeUtility กรุณาทำตามคำชี้แจงและข้อจำกัดด้านบน