I\'m trying to make this function accept a single element tensor.
def classi(i): out = np.zeros((1, 49), np.uint8) for j in range(len(classcount)):