fix diffedit code mistake
* part 2 of api fixes * move randn_tensor * add to toctree * apply feedback * more feedback