Thanks!
One more question:
If I want to create a tensor with the same options but with type int, is there something like
    std::vector<at::Tensor> forward(at::Tensor xy, int height, int width) {
        auto tmp  = at::empty({10, 4, height, width}, xy.options().as_int());
        return {tmp}
    }