fix device for input tensor (PyTorch)
Changing device for the input tensor was not added in the previous MR (!130 (merged)) which is added in this MR.
Changing device for the input tensor was not added in the previous MR (!130 (merged)) which is added in this MR.