Vector out of bounds error

[unnikrishnan-anil]

I was trying to run the code and I'm getting a vector out of bounds error in the following method in RayUtils.h:

inline torch::Tensor GetDirections(const int h, const int w, torch::Tensor k)
{
	auto device = k.device();
	//pytorch's meshgrid has indexing='ij'
	std::vector<torch::Tensor> ij = torch::meshgrid({ torch::linspace(0, w - 1, w), torch::linspace(0, h - 1, h) });
	auto i = ij[0].to(device).t();
	auto j = ij[1].to(device).t();
	torch::Tensor dirs = torch::stack({ (i - k[0][2]) / k[0][0], -(j - k[1][2]) / k[1][1], -torch::ones_like(i) }, -1);
	return dirs;
}

Seems to be on:

auto i = ij[0].to(device).t();

I am running this on a RTX 2080Ti. Do you think I'm running out of memory or something along those lines?