Added torch support

Signed-off-by: kecinzer <kecinzer@gmail.com>