Exemple #1
0
 def updateGrad(self, grad):
     self.grad = Memory.moveaxis(grad, self.dst, self.src)
Exemple #2
0
 def updateGrad(self, grad):
     self.grad = Memory.depthSplit(grad, self.inData)
Exemple #3
0
 def updateData(self, data):
     self.data = Memory.moveaxis(data, self.src, self.dst)
Exemple #4
0
 def updateData(self, data):
     self.data = Memory.depthConcat(data)
Exemple #5
0
 def updateGrad(self, grad):
     self.grad = Memory.swapaxes(grad, self.axis1, self.axis2)
Exemple #6
0
 def updateData(self, data):
     self.data = Memory.swapaxes(data, self.axis1, self.axis2)
Exemple #7
0
 def updateGrad(self, grad):
     self.grad = Memory.transpose(grad, self.invaxes)
Exemple #8
0
 def updateData(self, data):
     self.data = Memory.transpose(data, self.axes)