예제 #1
0
def test_stack_insufficient_inputs():
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([])
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([], 0)
예제 #2
0
 def forward_chainerx(self, xs):
     return chainerx.stack(xs, self.axis),
예제 #3
0
def test_stack_insufficient_inputs():
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([])
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([], 0)
예제 #4
0
파일: stack.py 프로젝트: ktnyt/chainer 
 def forward_chainerx(self, xs):
     return chainerx.stack(xs, self.axis),