def mnp_split(input_tensor):
a = mnp.split(input_tensor, indices_or_sections=1)
b = mnp.split(input_tensor, indices_or_sections=3)
c = mnp.split(input_tensor, indices_or_sections=(-9, -8, 6))
d = mnp.split(input_tensor, indices_or_sections=(3, 2, 1))
e = mnp.split(input_tensor, indices_or_sections=(-10, -4, 5, 10))
f = mnp.split(input_tensor, indices_or_sections=[0, 2], axis=1)
return a, b, c, d, e, f
def mnp_split(input_tensor):
a = mnp.split(input_tensor, indices_or_sections=1)
b = mnp.split(input_tensor, indices_or_sections=3)
return a, b
def split(x, indices_or_sections, axis=0):
return mnp.split(x, indices_or_sections, axis)