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按行求解
Tensorflow 函数汇总(持续更新)
猜你喜欢
转载自blog.csdn.net/qq_41686130/article/details/95814344
今日推荐
周排行