Home / Class/ filter Class — pytorch Architecture

filter Class — pytorch Architecture

Architecture documentation for the filter class in functional.h from the pytorch codebase.

Entity Profile

Source Code

aten/src/ATen/core/functional.h lines 37–47

template<typename F, typename T>
inline std::vector<T> filter(at::ArrayRef<T> inputs, const F& fn) {
  std::vector<T> r;
  r.reserve(inputs.size());
  for(auto & input : inputs) {
    if (fn(input)) {
      r.push_back(input);
    }
  }
  return r;
}

Analyze Your Own Codebase

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

Try Supermodel Free