给定三个正整数 n、m、h(均小于 100),构造如下数据并计算结果。 数据构造规则: 输入特征矩阵 X 为 n×m 的全 1 矩阵。 三个权重矩阵 W1、W2、W3 均为 m×h 的“上三角全 1”矩阵(按行列索引在主对角线及其上方位置为 1,其余为 0;当 m≠h 时视为按行列索引的上三角扩展)。 令 Q=X·W1,K=X·W2,V=X·W3;计算 S=(Q·K^T)sqrt(h)。 softmax 按行做“归一化”:对任意行向量 r,softmax(r) 的每个元素等于该元素除以本行所有元素之和。 Y=softmax(S)·V。 输出要求:求矩阵 Y 所有元素的和,四舍五入到整数后输出。
输入描述:
一行,三个正整数 n m h(均小于 100,且均0)
输出描述:
一行,一个整数:矩阵 Y 的元素和(四舍五入后)
示例1
说明
h≤m,单行和为 1+2+3=6;总和= n×6 = 5×6=30。
备注:
本题由牛友@Charles 整理上传
加载中...