import numpy as np
# 设置矩阵的维度
H = 5 # 行数
r = 3 # 列数
# 生成随机矩阵 A 和 B
A = np.random.rand(H, r)
B = np.random.rand(r, H)
A_normalized = A / np.sum(A, axis=0, keepdims=True)
B_normalized = B / np.sum(B, axis=0, keepdims=True)
# 计算矩阵 C = A_normalized @ B_normalized
C = A_normalized @ B_normalized
# 检查矩阵 C 是否归一化
C_norm = np.sum(C, axis=0)
# 打印结果
print("矩阵 C 的行范数:", C_norm)
print("矩阵 C 是否归一化:", np.allclose(C_norm, 1))