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