Skip to content

Commit 23d6eb8

Browse files
committed
Added const access operators
1 parent e2f7ffd commit 23d6eb8

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

src/autodiff.cc

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,10 @@ double& Var::operator()(size_t x, size_t y, size_t z, size_t t){
130130
return data(x,y,z,t);
131131
}
132132

133+
double Var::operator()(size_t x, size_t y, size_t z, size_t t) const {
134+
return data(x,y,z,t);
135+
}
136+
133137
Var Var::operator+(const Var& y)const{
134138
auto new_data = data + y.data;
135139
auto x_weight = nn::Tensor(data.shape, 1);

src/autodiff.hpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ class Var{
3333
void evaluate_leaves() const;
3434
nn::Tensor grad()const;
3535

36-
double& operator()(size_t,size_t=1,size_t=1,size_t=1);
36+
double& operator()(size_t, size_t=0, size_t=0, size_t=0);
37+
double operator()(size_t, size_t=0, size_t=0, size_t=0) const;
3738

3839
Var operator+(const Var&)const;
3940
Var operator-(const Var&)const;

0 commit comments

Comments
 (0)