Example #1
0
def test_stack_insufficient_inputs():
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([])
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([], 0)
Example #2
0
 def forward_chainerx(self, xs):
     return chainerx.stack(xs, self.axis),
Example #3
0
def test_stack_insufficient_inputs():
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([])
    with pytest.raises(chainerx.DimensionError):
        chainerx.stack([], 0)
Example #4
0
 def forward_chainerx(self, xs):
     return chainerx.stack(xs, self.axis),