def test_stack_insufficient_inputs(): with pytest.raises(chainerx.DimensionError): chainerx.stack([]) with pytest.raises(chainerx.DimensionError): chainerx.stack([], 0)
def forward_chainerx(self, xs): return chainerx.stack(xs, self.axis),