reduction_ops_test.py 文件源码

def testComplexReduce3D(self):
    # Create a 3D array of floats and reduce across all possible
    # dimensions
    np_arr = (np.linspace(10, -10, 30) +
              1j*np.linspace(-10, 10, 30)).reshape(
                [2, 3, 5]).astype(np.complex64)
    self._compareAll(np_arr, None)
    self._compareAll(np_arr, [])
    self._compareAll(np_arr, [0])
    self._compareAll(np_arr, [1])
    self._compareAll(np_arr, [2])
    self._compareAll(np_arr, [0, 1])
    self._compareAll(np_arr, [1, 2])
    self._compareAll(np_arr, [0, 2])
    self._compareAll(np_arr, [0, 1, 2])