def backward_gpu(self, x, gy): gx = cuda.elementwise( 'T x, T gy', 'T gx', 'gx = abs(x) > 1 ? 0 : gy', 'bst_bwd')( x[0], gy[0]) return gx,