From the sound of you being able to let it cool down and drive it further, that might be a bad input piston seal (common enough) or a clogged filter. Either way, both would require the trans come apart and be gone through for nearly a full rebuild.