Ejemplos de split en Python

Ejemplo n.º 1

 def forward_chainerx(self, inputs):
     x, = inputs
     return tuple(chainerx.split(x, self.indices_or_sections, self.axis))

Ejemplo n.º 2

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)

Ejemplo n.º 3

Archivo: test_manipulation.py Proyecto: pfnet/chainer

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)

Ejemplo n.º 4

Archivo: split_axis.py Proyecto: ktnyt/chainer

 def forward_chainerx(self, inputs):
     x, = inputs
     return tuple(chainerx.split(x, self.indices_or_sections, self.axis))