def upper_tri_vec(matrix):
"Returns the vectorized values of upper triangular part of a matrix"
triu_idx = np.triu_indices_from(matrix, 1)
return matrix[triu_idx]
微信
公众号
扫码关注公众号