Got it . just one more question and this is not related to the topic . what data types and operations  are allowed when pytorch is working with cuda .I couldnt find anything specific  in pytorch documentation .Beacuse these  instructions and data  type have to be loaded into gpu . like can I use if statements or external functions like  some other  (ofcourse it has to already already be there in the gpu). or can I use sets, dictionaries , tuples, numpy arrays  , …in my model, when later on trying to use the .to(‘cuda’) to load the model to gpu ? ( Or have I got it all wrong . if you could share  a link conected to this topic  I would  really appriciate it )