Nabla  1.0
Nabla - a DSL for Automatic differentiation
Public Member Functions | Public Attributes | List of all members
Transpose Class Reference

#include <Operators.h>

Inheritance diagram for Transpose:
Inheritance graph
[legend]
Collaboration diagram for Transpose:
Collaboration graph
[legend]

Public Member Functions

void backward ()
 
 Transpose (Node *a, int count)
 
Nodeforward (const Node *a)
 
- Public Member Functions inherited from Operator
 Operator ()
 
- Public Member Functions inherited from Node
 Node ()
 
virtual ~Node ()=default
 
void print ()
 
 Node ()
 
 Node (Tensor &data)
 
Nodeforward (const Node &a, const Node &b)
 

Public Attributes

int trans_count
 
- Public Attributes inherited from Node
std::string name
 
bool is_visited = false
 
bool is_scalar = false
 
std::vector< Node * > inputs
 
Tensor data
 
double ddata
 
Tensor gradient
 
double scalar_gradient = 0
 

Constructor & Destructor Documentation

◆ Transpose()

Transpose::Transpose ( Node a,
int  count 
)

Member Function Documentation

◆ backward()

void Transpose::backward ( )
virtual

Reimplemented from Operator.

◆ forward()

Node * Transpose::forward ( const Node a)

Member Data Documentation

◆ trans_count

int Transpose::trans_count

The documentation for this class was generated from the following files: