torch.Tensor.take\_along\_dim ============================= .. currentmodule:: torch .. automethod:: Tensor.take_along_dim