Home / Class/ map Class — pytorch Architecture

map Class — pytorch Architecture

Architecture documentation for the map class in NestedTensorUtils.h from the pytorch codebase.

Entity Profile

Source Code

aten/src/ATen/native/nested/NestedTensorUtils.h lines 337–346

template <class F, class... B>
static inline NestedNode<
    typename c10::guts::infer_function_traits<F>::type::return_type>
map(F&& fn, const NestedNode<B>&... nested_node) {
  return _map<
      F,
      typename c10::guts::infer_function_traits<F>::type::return_type,
      typename c10::guts::infer_function_traits<F>::type::parameter_types>::
      function(std::forward<F>(fn), nested_node...);
}

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free