for slice in inputTensor.axis(0):
  if slice[0, dotProductColumn] >= 0:
    a.add(slice)
  else:
    b.add(slice)