Use torch.equal for tensor equality comparisons

VykosX / ControlFlowUtils

Custom nodes for ComfyUI to enable flow control with advanced loops, conditional branching, logic operations and several other nifty utilities to enhance your ComfyUI workflows

GNU General Public License v3.0

70 stars 4 forks source link

Use torch.equal for tensor equality comparisons #2

Closed christian-byrne closed 1 month ago

christian-byrne commented 1 month ago

torch.equal handles equality better than torch.Tensor.__eq__ for this purpose. In particular, it won't raise exception when tensors have different rank or shape.

VykosX commented 1 month ago

Thank you for your contribution! Merged.