Exemplo n.º 1
0
 def forward(self, y):
     return tt.advanced_set_subtensor1(y, tt.log(y[self.diag_idxs]), self.diag_idxs)
Exemplo n.º 2
0
 def forward(self, y):
     return tt.advanced_set_subtensor1(y, tt.log(y[self.diag_idxs]), self.diag_idxs)
Exemplo n.º 3
0
 def backward(self, x):
     return tt.advanced_set_subtensor1(x, tt.exp(x[self.diag_idxs]), self.diag_idxs)
Exemplo n.º 4
0
 def backward(self, x):
     return tt.advanced_set_subtensor1(x, tt.exp(x[self.diag_idxs]), self.diag_idxs)