optimize_torch

Hint

Modified objects to fit the Torch tensor calculation.