Operator not supported

General
1

Hello, this line in my code generates the unsupported ScatterND operator. I changed it to torch.cat, etc., but it’s still not supported. Do you have any good solutions?

output[..., VX:] = velocity + anchor[..., VX:]
3

Hey @chenyao ,

The reason you get ScatterND is because of the in-place update. Try to rebuild the whole tensor instead of modifying it.

Try this:

left  = anchor[..., :VX]
right = anchor[..., VX:] + velocity
output = torch.cat([left, right], dim=-1)

Hope this helps

1 Like