コード例 #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)