Python separateの例

コード例 #1

ファイル: test_separate.py プロジェクト: PaulGureghian1/Chainer

    def check_forward(self, x_data):
        x = chainer.Variable(x_data)
        ys = separate.separate(x, self.axis)

        self.assertIsInstance(ys, tuple)
        self.assertEqual(len(ys), self.shape[self.axis])
        for i in range(self.shape[self.axis]):
            expect = self.x.take(i, axis=self.axis)
            testing.assert_allclose(ys[i].data, expect)

コード例 #2

ファイル: test_separate.py プロジェクト: fukatani/chainer

    def check_forward(self, x_data):
        x = chainer.Variable(x_data)
        ys = separate.separate(x, self.axis)

        self.assertIsInstance(ys, tuple)
        self.assertEqual(len(ys), self.shape[self.axis])
        for i in range(self.shape[self.axis]):
            expect = self.x.take(i, axis=self.axis)
            testing.assert_allclose(ys[i].data, expect)

コード例 #3

ファイル: test_separate.py プロジェクト: PaulGureghian1/Chainer

 def f(x):
     return separate.separate(x, self.axis)

コード例 #4

ファイル: test_separate.py プロジェクト: fukatani/chainer

 def f(x):
     return separate.separate(x, self.axis)