|
|
| Variable (const tensor::Tensor &data, bool requires_grad=false) |
| |
|
const tensor::Tensor & | grad () const |
| |
|
const tensor::Tensor & | data () const noexcept |
| |
|
std::weak_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 () |
| |
The documentation for this class was generated from the following file: