我正在使用Keras和Tensorflow构建一个自定义层,用于计算一组框之间的相交。A有两组锚点和具有不同内容的gt_boxes,在执行以下错误代码时,我将计算锚中每个元素与gt_boxes之间的交集:Tracebackassert_is_compatible_with
ValueError: Dimensions 3 and 10 are not compatib
我想将一个大小为(n X N X M X m)的张量T转换为一个大小为(n X M X m)的张量U,同时只检索(NxN)块(即Uikl=Tiikl)上T的对角元素。torch.diag()只适用于2-D张量,我真的看不出如何在不对元素的索引进行循环的情况下做到这一点(考虑到我认为它的计算效率很低,我希望避免这种情况)。rang