Tensorflow 函数汇总(持续更新)

sess = tf.InteractiveSession()

arr = np.array([[31, 23,  4, 24, 27, 34],
                [18,  3, 25,  0,  6, 35],
                [28, 14, 33, 22, 20,  8],
                [13, 30, 21, 19,  7,  9],
                [16,  1, 26, 32,  2, 29],
                [17, 12,  5, 11, 10, 15]])
print('矩阵维度')
print(tf.rank(arr).eval())
print('矩阵行列')
print(tf.shape(arr).eval())
print('返回最大值的索引')
print(tf.argmax(arr, 1).eval())#0按列求解,1按行求解

猜你喜欢

转载自blog.csdn.net/qq_41686130/article/details/95814344