Exemplo n.º 1
0
def pmap1(f, x, w = 3):
  n = x.shape[0]
  def local_apply(i):
    lower = __builtins__.max(i-w/2, 0)
    upper = __builtins__.min(i+w/2+1, n)
    elts = x[lower:upper]
    return f(elts)
  return imap(local_apply, n)
Exemplo n.º 2
0
def pmap1(f, x, w=3):
    n = x.shape[0]

    def local_apply(i):
        lower = __builtins__.max(i - w / 2, 0)
        upper = __builtins__.min(i + w / 2 + 1, n)
        elts = x[lower:upper]
        return f(elts)

    return imap(local_apply, n)
Exemplo n.º 3
0
def pmap2(f, x, width=(3, 3)):
    """
  Patch-map where the function can accept both interior windows
  and smaller border windows
  """
    width_x, width_y = width
    n_rows, n_cols = x.shape
    hx = width_x / 2
    hy = width_y / 2

    def local_apply((i, j)):
        lx = __builtins__.max(i - hx, 0)
        ux = __builtins__.min(i + hx + 1, n_rows)
        ly = __builtins__.max(j - hy, 0)
        uy = __builtins__.min(j + hy + 1, n_cols)
        return f(x[lx:ux, ly:uy])

    return imap(local_apply, x.shape)
Exemplo n.º 4
0
def pmap2(f, x, width = (3,3)):
  """
  Patch-map where the function can accept both interior windows
  and smaller border windows
  """
  width_x, width_y = width 
  n_rows, n_cols = x.shape
  hx = width_x / 2
  hy = width_y / 2
  
  def local_apply((i,j)):
    lx = __builtins__.max(i-hx, 0)
    ux = __builtins__.min(i+hx+1, n_rows)
    ly = __builtins__.max(j-hy, 0)
    uy = __builtins__.min(j+hy+1, n_cols)
    return f(x[lx:ux, ly:uy])
    
  return imap(local_apply, x.shape)
Exemplo n.º 5
0
def ones(shape, dtype = float64):
  one = dtype(1)
  return imap(lambda _: one, shape)
Exemplo n.º 6
0
def zeros(shape, dtype = float64):
  zero = dtype(0)
  return imap(lambda _: zero, shape)