Python stackの例

コード例 #1

ファイル: test_manipulation.py プロジェクト: pfnet/chainer

def test_stack_insufficient_inputs():
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([])
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([], 0)

コード例 #2

 def forward_chainerx(self, xs):
     return chainerx.stack(xs, self.axis),

コード例 #3

def test_stack_insufficient_inputs():
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([])
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([], 0)

コード例 #4

ファイル: stack.py プロジェクト: ktnyt/chainer

 def forward_chainerx(self, xs):
     return chainerx.stack(xs, self.axis),