根据错误提示,问题出在以下line 167行代码:
indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1)))
错误提示指出无法将浮点数类型(Float)转换为所需的输出类型(__int64)
indices.append((b, a, gj.clamp(0, int(gain[3]) - 1), gi.clamp(0, int(gain[2]) - 1)))
这样修改后,应该就能够正确运行代码了