def forward_chainerx(self, inputs): x, = inputs return tuple(chainerx.split(x, self.indices_or_sections, self.axis))
def test_split_zero_sized_no_offset(device, shape, indices_or_sections, axis): # An (sub-)array of size 0 should always have 0 offset. a = chainerx.random.uniform(-1, 1, shape) assert a.offset == 0 # Test pre-condition. b = chainerx.split(a, indices_or_sections, axis) assert all(bi.offset == 0 for bi in b)
def test_split_zero_sized_no_offset(device, shape, indices_or_sections, axis): # An (sub-)array of size 0 should always have 0 offset. a = chainerx.random.uniform(-1, 1, shape) assert a.offset == 0 # Test pre-condition. b = chainerx.split(a, indices_or_sections, axis) assert all(bi.offset == 0 for bi in b)
def forward_chainerx(self, inputs): x, = inputs return tuple(chainerx.split(x, self.indices_or_sections, self.axis))