Again for Torch

Change-Id: I45c577cd32d8d0fb3cf7de25641e2dd4409031c1