To make sure I understand correctly: This is extracting the diagonal of the local tensor of a given "original" kernel?