Python join_pretty_tensors 예제들

예제 #1

    def testDefaultJoinAlignmentError(self):
        pt1 = self.Wrap(numpy.array([[1.0], [2.0]]))
        pt2 = self.Wrap(numpy.array([[2.0, 3.0], [4.0, 5.0]]))
        pt3 = self.Wrap(numpy.array([5.0, 6.0]))

        with self.assertRaises(ValueError):
            prettytensor.join_pretty_tensors([pt1, pt2, pt3], pt1)

예제 #2

파일: pretty_tensor_test.py 프로젝트: roger2000hk/prettytensor

  def testDefaultJoinAlignmentError(self):
    pt1 = self.Wrap(numpy.array([[1.0], [2.0]]))
    pt2 = self.Wrap(numpy.array([[2.0, 3.0], [4.0, 5.0]]))
    pt3 = self.Wrap(numpy.array([5.0, 6.0]))

    with self.assertRaises(ValueError):
      prettytensor.join_pretty_tensors([pt1, pt2, pt3], pt1)

예제 #3

파일: pretty_tensor_test.py 프로젝트: pombredanne/prettytensor

    def testCustomJoin(self):
        pt1 = self.Wrap(numpy.array([[1.0], [2.0]]))
        pt2 = self.Wrap(numpy.array([[2.0], [4.0]]))
        pt3 = self.Wrap(numpy.array([[5.0], [6.0]]))
        out_pt = prettytensor.join_pretty_tensors([pt1, pt2, pt3], pt1, join_function=tf.add_n)

        out = self.RunTensor(out_pt)

        testing.assert_allclose(out, numpy.array([[8.0], [12.0]]), rtol=TOLERANCE)

예제 #4

파일: pretty_tensor_test.py 프로젝트: pombredanne/prettytensor

    def testDefaultJoinFunction(self):
        pt1 = self.Wrap(numpy.array([[1.0], [2.0]]))
        pt2 = self.Wrap(numpy.array([[2.0, 3.0], [4.0, 5.0]]))
        pt3 = self.Wrap(numpy.array([[5.0], [6.0]]))
        out_pt = prettytensor.join_pretty_tensors([pt1, pt2, pt3], pt1)

        out = self.RunTensor(out_pt)

        testing.assert_allclose(out, numpy.array([[1.0, 2.0, 3.0, 5.0], [2.0, 4.0, 5.0, 6.0]]), rtol=TOLERANCE)

예제 #5

파일: pretty_tensor_test.py 프로젝트: sskgit/prettytensor

  def testCustomJoin(self):
    pt1 = self.Wrap(numpy.array([[1.0], [2.0]]))
    pt2 = self.Wrap(numpy.array([[2.0], [4.0]]))
    pt3 = self.Wrap(numpy.array([[5.0], [6.0]]))
    out_pt = prettytensor.join_pretty_tensors(
        [pt1, pt2, pt3], pt1, join_function=tf.add_n)

    out = self.RunTensor(out_pt)

    testing.assert_allclose(out, numpy.array([[8.0], [12.0]]), rtol=TOLERANCE)

예제 #6

파일: pretty_tensor_test.py 프로젝트: sskgit/prettytensor

  def testDefaultJoinFunction(self):
    pt1 = self.Wrap(numpy.array([[1.0], [2.0]]))
    pt2 = self.Wrap(numpy.array([[2.0, 3.0], [4.0, 5.0]]))
    pt3 = self.Wrap(numpy.array([[5.0], [6.0]]))
    out_pt = prettytensor.join_pretty_tensors([pt1, pt2, pt3], pt1)

    out = self.RunTensor(out_pt)

    testing.assert_allclose(
        out, numpy.array([[1.0, 2.0, 3.0, 5.0], [2.0, 4.0, 5.0, 6.0]]),
        rtol=TOLERANCE)