我试图将矩阵的严格上三角部分转换为Tensorflow中的数组。这是一个例子:输入:[[1,2,3],[4,5,6],[7,8,9]]输出:[2,3,6]我尝试了下面的代码，但没有成功(报错):defupper_triangular_to_array(A):mask=tf.matrix_band_part(tf.ones_like(A,dtype=tf.bool),0,-1)returntf.boolean_mask(A,mask)谢谢! 最佳答案 以下答案与@Cech_Cohomology的答案非常接近，但它在过程中不使用Nump