fix device for input tensor

4 jobs for fix-pytorch in 48 minutes and 57 seconds (queued for 8 seconds)
Status Job ID Name Coverage
  Build
passed #240529
docker
build_linux_37

00:48:57

66.0%
passed #240530
docker
build_linux_38

00:46:38

67.0%
passed #240527
intel macos
build_macos_intel_37

00:19:21

58.0%
passed #240528
intel macos
build_macos_intel_38

00:16:51

59.0%