题目
样例输入
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
代码
n,d=map(int,input().split(' '))#输入3 2
#map() 会根据提供的函数对指定序列做映射
#n=int(n)
#d=int(d)
#n行d列
#QKV
Q=[]
K=[]
V=[]
W=[]
for i in range(n):#######退回缩进ctrl+[
Q.append(list(map(int,input().split(' '))))
for i in range(n):#######退回缩进ctrl+[
K.append(list(map(int,input().split(' '))))
for i in range(n):#######退回缩进ctrl+[
V.append(list(map(int,input().split(' '))))
W=(list(map(int,input().split(' '))))
##计算转置
KT=[]
for t in range(d):
temp=[]
for tt in K:
temp.append(int(tt[t]))
KT.append(temp)
def ss(left, right, a, b, c):
# left:a*b right:b*c return ans
ans = [[0]*c for i in range(a)]
for i in range(a):
for j in range(c):
# get ans[i][j]
temp = 0
for k in range(b):
temp += left[i][k]*right[k][j]
ans[i][j]=temp
return ans
#矩阵相乘
#KT*V
asn=ss(KT,V,d,n,d)
result=ss(Q,asn,n,d,d)
for i in range (n):
for j in range (d):
result[i][j]*=W[i]
for i in range(n):
for j in range(d):
print(result[i][j], end=' ')
print()
注意
代码优化
1先算KT X V得到d X d 的矩阵,再算Q X KT 得到n X d的矩阵
2矩阵计算用函数可以节省大量内存!!!!
3map() 会根据提供的函数对指定序列做映射
矩阵计算代码
def ss(left, right, a, b, c):#a行b列 left X b行c列 right
ans = [[0]*c for i in range(a)]
for i in range(a):
for j in range(c):
# get ans[i][j]
temp = 0
for k in range(b):
temp += left[i][k]*right[k][j]
ans[i][j]=temp
return ans
print(result[i][j], end=' ')# 以空格为分隔符,不换行,但是最后有空格