Home / Class/ bernoulli_impl_ Class — pytorch Architecture

bernoulli_impl_ Class — pytorch Architecture

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

Class c

Entity Profile

Source Code

aten/src/ATen/native/DistributionTemplates.h lines 362–369

template<template<typename> class bernoulli_tensor_kernel, typename RNG>
Tensor& bernoulli_impl_(Tensor& self, const Tensor& p_, std::optional<Generator> gen) {
  CHECK_EMPTY_AND_RETURN(self);
  NoNamesGuard guard;
  at::assert_no_internal_overlap(self);
  bernoulli_tensor_kernel<RNG>()(self, p_, gen);
  return self;
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free