Is var[0,:] unpacking for using autograd based backprop or not?

So Is there any way to manipulate some parts of my output variable and avoid this error? Because as you can see in My loss function, I should take square of some parts of output variable!