def kos_in_gen(generated, gen_idx, all_kos):
	"""
	Return list genes/KOs encoded by a generated genome vector
	
	Arguments:
		generated (tensor) -- generated genome vectors. Rows are genomes, columns are genes. 1's denote a gene is encoded, 0 denotes that it is not
		gen_idx (int) -- index of generated genome in set of real + generated genomes
		all_kos (list) -- list of all KOs in the dataset
		
	Returns:
		gen_kos (list) -- KO numbers encoded by genome vector
	"""
	gen_ko_idx = [int(i) for i in (generated[gen_idx] == 1).nonzero()]
	gen_kos = [all_kos[i] for i in gen_ko_idx]
	print("There are a total of",len(gen_kos),"genes encoded in this genome vector")
	
	return gen_kos