Does pytorch have a function similar to tensorflow?

vision
1

Such as map_fn, scatter_nd, gather_nd.thanks

2

The pytorch equivalent for the above are map_, scatter_, gather respectively.
you can have a look at - https://pytorch.org/docs/stable/tensors.html#torch.Tensor.scatter_