|
| Variable (const tensor::Tensor &data, bool requires_grad=false) |
|
const tensor::Tensor & | grad () const noexcept |
|
const tensor::Tensor & | data () const noexcept |
|
const std::shared_ptr< Function > & | grad_fn () const noexcept |
|
bool | requires_grad () const noexcept |
|
void | zero_grad () |
|
void | incr_grad (const tensor::Tensor &other_grad) |
|
void | set_grad_fn (std::shared_ptr< Function > grad_fn) |
|
void | copy (const Variable &other) |
|
void | fill (tensor::Scalar item) |
|
void | backward () |
|
|
std::ostream & | operator<< (std::ostream &os, const Variable &obj) |
|
The documentation for this class was generated from the following files:
- include/lamppp/autograd/variable.hpp
- src/autograd/variable.cpp