Modificiation to how stack output is chosen
Modificiation to how stack output is chosen

yozw/torch-totem,deepmind/torch-totem,dm-jrae/torch-totem,fzvinicius/torch-totem,mevGDM/torch-totem,GeorgOstrovski/torch-totem,clementfarabet/torch-totem,fbesse/torch-totem