Tensorflow vs Pytorch 명령어 비교
tf.argsort(values, axis=-1, direction='ASCENDING') arr = [1, 2, 3, 10, 5, 6] arr_ascending = tf.argsort(arr, axis=-1, direction='ASCENDING') arr_descending = tf.argsort(arr, axis=-1, direction='DESCENDING') with tf.Session() as sess: print(sess.run(arr_ascending)) print(sess.run(arr_descending)) # ----------------------------------- arr2 = [[1, 2], [4, 8], [5, 3], [2, 7]] print(np.array(arr2).sh..
2021. 5. 14.