def METHOD_NAME(self, input_np, block_size, mode):
    a_np = np.transpose(input_np, axes=[0, 3, 1, 2])
    ref_np = depth_to_space_python(a_np, block_size, mode=mode)
    ref_np = np.transpose(ref_np, axes=[0, 2, 3, 1])
    return ref_np