arr0 = tf.range(12)
arr1 = tf.random.shuffle(arr0)
arr = tf.reshape(arr1, [3, 4])
t11 = tf.reduce_min(arr)
t12 = tf.reduce_min(arr, axis=1)
t21 = tf.reduce_max(arr)
t22 = tf.reduce_max(arr, axis=1)
t31 = tf.reduce_mean(arr)
t32 = tf.reduce_mean(arr, axis=1)
print(arr)
print(tf.argmin(arr))
print(tf.argmax(arr))
print(tf.argmax(arr))
print(t11)
print(t12)
print(tf.argmin(arr, axis=1))
print(t21)
print(t22)
print(tf.argmax(arr, axis=1))
print(t31)
print(t32)
print(tf.argmax(arr, axis=1))
近期评论