Home / Class/ exponential_impl_ Class — pytorch Architecture

exponential_impl_ Class — pytorch Architecture

Architecture documentation for the exponential_impl_ class in DistributionTemplates.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/DistributionTemplates.h lines 337–344

template<template<typename> class exponential_kernel, typename RNG>
Tensor& exponential_impl_(Tensor& self, double lambda, std::optional<Generator> gen) {
  TORCH_CHECK(lambda > 0.0, "exponential_ expects lambda > 0.0, but found lambda=", lambda);
  CHECK_EMPTY_AND_RETURN(self);
  auto iter = TensorIterator::borrowing_nullary_op(self);
  exponential_kernel<RNG>()(iter, lambda, gen);
  return self;
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free