Home / Class/ map_nested_tensor Class — pytorch Architecture

map_nested_tensor Class — pytorch Architecture

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

Class c

Entity Profile

Source Code

aten/src/ATen/native/nested/NestedTensorUtils.h lines 453–461

template <class F, class... A>
inline at::Tensor map_nested_tensor(F&& fn, A... a) {
  return wrap_tensor_node(
      impl::map(std::forward<F>(fn), impl::get_nested_tensor_structure(a)...),
      std::nullopt,
      std::nullopt,
      std::nullopt,
      std::nullopt);
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free