예제 #1
0
def _full_like(x, val):
    if isinstance(x, numpy.ndarray):
        return numpy.full_like(x, val)
    else:
        return cuda.full_like(x, val)
예제 #2
0
파일: sum.py 프로젝트: nihohi0428/chainer 
 def backward_gpu(self, x, gy):
     # TODO(beam2d): Make it async
     return cuda.full_like(x[0], gy[0].get()),
예제 #3
0
def _full_like(x, val):
    if isinstance(x, numpy.ndarray):
        return numpy.full_like(x, val)
    else:
        return cuda.full_like(x, val)