Function at::take_along_dim#

• Defined in File Functions.h

Function Documentation#

inline at::Tensor at::take_along_dim(const at::Tensor &self, const at::Tensor &indices, ::std::optional<int64_t> dim = ::std::nullopt)#