Home / Class/ map_nt Class — pytorch Architecture

map_nt Class — pytorch Architecture

Architecture documentation for the map_nt class in NestedTensorMath.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/nested/NestedTensorMath.h lines 14–19

template <typename Func>
Tensor map_nt(const Tensor& nt, Func f) {
  auto* nt_impl = get_nested_tensor_impl(nt);
  const auto& sizes = nt_impl->get_nested_sizes();
  return at::detail::make_tensor<NestedTensorImpl>(f(nt_impl->get_buffer()), sizes);
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free