Fix import for GpuSubtensor.