Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
torch::autograd::Edge Struct Reference

Represents a particular input of a function. More...

#include <edge.h>

Public Member Functions

 Edge (std::shared_ptr< Function > function_, uint32_t input_nr_) noexcept
 
bool is_valid () const noexcept
 Convenience method to test if an edge is valid.
 
bool operator== (const Edge &other) const noexcept
 
bool operator!= (const Edge &other) const noexcept
 

Data Fields

std::shared_ptr< Function > function
 The function this Edge points to.
 
uint32_t input_nr
 The identifier of a particular input to the function.
 

Detailed Description

Represents a particular input of a function.

Definition at line 14 of file edge.h.


The documentation for this struct was generated from the following file: