prepare_data_torch function

Function to additionally prepare data for fit process (torch)