connected-components-3d
connected-components-3d copied to clipboard
Any way to make this GPU Compatible?
Maybe a naive question, but -> It becomes a bit tedious detaching and reattaching stuff when this is required within a pipeline. Is there any way this can be done through something like Pytorch?