Python map_fn 예제들

예제 #1

파일: map_test.py 프로젝트: yz3052/tensorflow-upstream

  def testShortCircuit(self, structure, map_fn, num_parallel_calls):
    dataset = self.structuredDataset(structure).repeat().map(
        map_fn, num_parallel_calls=num_parallel_calls)
    get_next = self.getNext(dataset)

    if isinstance(structure, tuple):
      expected = map_fn(*self.evaluate(self.structuredElement(structure)))
    else:
      expected = map_fn(self.evaluate(self.structuredElement(structure)))
    self.assertEqual(expected, self.evaluate(get_next()))

예제 #2

파일: map_test.py 프로젝트: perfmjs/tensorflow

  def testShortCircuit(self, structure, map_fn, num_parallel_calls):
    dataset = self.structuredDataset(structure).repeat().map(
        map_fn, num_parallel_calls=num_parallel_calls)
    get_next = self.getNext(dataset)

    if isinstance(structure, tuple):
      expected = map_fn(*self.evaluate(self.structuredElement(structure)))
    else:
      expected = map_fn(self.evaluate(self.structuredElement(structure)))
    self.assertEqual(expected, self.evaluate(get_next()))