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