Method testArgsort

tensorflow/python/ops/sort_ops_test.py:127–133

(self)

  @test_util.run_deprecated_v1
  def testArgsort(self):
    arr = np.random.random((5, 6, 7, 8))
    for axis in range(4):
      with self.cached_session():
        self.assertAllEqual(
            np.argsort(arr, axis=axis),
            sort_ops.argsort(arr, axis=axis).eval())


if __name__ == '__main__':