priors = x[None, :, :, None, :] @ self.route_weights[:, None, :, :, :]

这里的@代表矩阵乘法：

A = [[1, 2],    B = [[11, 12],
     [3, 4]]         [13, 14]]

# 元素乘法会产生
A * B = [[1 * 11,   2 * 12], 
         [3 * 13,   4 * 14]]

# 矩阵乘法会产生；@与np.matmul()相当
A @ B  =  [[1 * 11 + 2 * 13,   1 * 12 + 2 * 14],
           [3 * 11 + 4 * 13,   3 * 12 + 4 * 14]]