torch.take\_along\_dim ====================== .. currentmodule:: torch .. autofunction:: take_along_dim