def _full_like(x, val):
    if isinstance(x, numpy.ndarray):
        return numpy.full_like(x, val)
    else:
        return cuda.full_like(x, val)
Example #2
0
 def backward_gpu(self, x, gy):
     # TODO(beam2d): Make it async
     return cuda.full_like(x[0], gy[0].get()),
Example #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)