Add torch back

Change-Id: I01a2958106538e8023006af87a0bb1be39bf1f9b
Signed-off-by: Dirk Rettschlag <dirk.rettschlag@gmail.com>