An error occurred in the kroneker_matrix function

[FirstDiscoverer]

The error is:

OP_REQUIRES failed at gather_nd_op.cc:47 : Invalid argument: indices[82,82] = [22, 22] does not index into param shape [17,17]

The number of lines where the code error occurred is this

The location of the function call is here

Looks like it's looking for an index error in self.K_D_v. Because the shape of self.K_D_v is [22, 22], but the searched index exceeds 22, why such an error occurs?

[FirstDiscoverer]

I found an error, it was caused by an incorrect value here