插入在tensorflow张量

Question

平均行我在numpy的以下函数：

A = np.array([[1, 2, 3], 
      [4, 5, 6], 
      [7, 8, 9], 
      [10, 11, 12]]) 


def insert_row_averages(A, n=2): 
    slice2 = A[n::n] 
    v = (A[n-1::n][:slice2.shape[0]] + slice2)/2.0 
    np.insert(A.astype(float),n*np.arange(1,v.shape[0]+1),v,axis=0) 

基本上取的上方和下方的行的平均，并将其插入在两者之间，在每n个间隔。

有关如何在Tensorflow中执行此操作的任何想法？具体来说，我可以使用什么来代替np.insert，因为似乎没有相同的功能。

Answer 1

你可以使用一个初始化为基础的方法来解决这个问题，像这样 -

def insert_row_averages_iniitalization_based(A, n=1): 

    # Slice and compute averages 
    slice2 = A[n::n] 
    v = (A[n-1::n][:slice2.shape[0]] + slice2)/2.0 

    # Compute number of rows for o/p array 
    nv = v.shape[0] 
    nrows = A.shape[0] + nv 

    # Row indices where the v values are the inserted in o/p array 
    v_rows = (n+1)*np.arange(nv)+n 

    # Initialize o/p array 
    out = np.zeros((nrows,A.shape[1])) 

    # Insert/assign v in output array 
    out[v_rows] = v # Assign v 

    # Create 1D mask of length equal to no. of rows in o/p array, such that its 
    # TRUE at places where A is to be assigned and FALSE at places where values 
    # from v were assigned in previous step. 
    mask = np.ones(nrows,dtype=bool) 
    mask[v_rows] = 0 

    # Use the mask to assign A. 
    out[mask] = A 
    return out