#P16668. [CSPro 30] 矩阵运算

    ID: 18753 远端评测题 1000ms 512MiB 尝试: 0 已通过: 0 显示难度普及− 上传者: 标签>模拟2023矩阵运算矩阵乘法CSPro

[CSPro 30] 矩阵运算

背景

洛谷的测试数据仅供民间交流使用,非官方测试数据。官方评测链接:https://www.cspro.org/。

题目描述

$\text{Softmax}(\frac{Q \times K^T}{\sqrt{d}}) \times V$ 是 Transformer 中注意力模块的核心算式,其中 QQ、KK 和 VV 均是 nn 行 dd 列的矩阵,KTK^T 表示矩阵 KK 的转置,×\times 表示矩阵乘法。

为了方便计算,顿顿同学将 Softmax 简化为了点乘一个大小为 nn 的一维向量 WW:

(W⋅(Q×KT))×V(W \cdot (Q \times K^T)) \times V

点乘即对应位相乘,记 W(i)W^{(i)} 为向量 WW 的第 ii 个元素,即将 (Q×KT)(Q \times K^T) 第 ii 行中的每个元素都与 W(i)W^{(i)} 相乘。

现给出矩阵 QQ、KK 和 VV 和向量 WW,试计算顿顿按简化的算式计算的结果。

输入格式

从标准输入读入数据。

输入的第一行包含空格分隔的两个正整数 nn 和 dd,表示矩阵的大小。

接下来依次输入矩阵 QQ、KK 和 VV。每个矩阵输入 nn 行,每行包含空格分隔的 dd 个整数,其中第 ii 行的第 jj 个数对应矩阵的第 ii 行、第 jj 列。

最后一行输入 nn 个整数,表示向量 WW。

输出格式

输出到标准输出。

输出共 nn 行,每行包含空格分隔的 dd 个整数,表示计算的结果。

3 2
1 2
3 4
5 6
10 10
-20 -20
30 30
6 5
4 3
2 1
4 0 -5
480 240
0 0
-2200 -1100

提示

子任务

70%70\% 的测试数据满足:n≤100n \le 100 且 d≤10d \le 10;输入矩阵、向量中的元素均为整数,且绝对值均不超过 3030。

全部的测试数据满足:n≤104n \le 10^4 且 d≤20d \le 20;输入矩阵、向量中的元素均为整数,且绝对值均不超过 10001000。

提示

请谨慎评估矩阵乘法运算后的数值范围,并使用适当数据类型存储矩阵中的整数。