cianfen的部落格

python re-order array or select specie column or row

    sel=[8,7,6,5,4,3,2,1]  #先選好指定的行數
    out=[0,9] #先選好指定的行數
    datain=data[:,sel]
    dataout=data[:,out]

    data_newR_in=data[sel,:]
    data_newR_out=data[out,:]

datain[i:(i+seq_length), 0:insize]  # 不含insize , 也不含(i+length) , datain.shape=[seg_length, insize]

pytorch:

        indexA = torch.tensor([0, 1, 2, 3,4,5,6,7,8])
        indexB = torch.tensor([8,9])

        in = torch.index_select(data, 0, indexA).float()
        tt = torch.index_select(data, 0, indexB).float()