np.concatenate()

np.concatenate()

np.concatenate()的用法和torch.cat()的用法相同

    qf.append(features)
    pids = pids.numpy().astype(np.int16)
    camids = camids.numpy().astype(np.int16)
    q_pids.append(pids)
    q_camids.append(camids) 
qf = torch.cat(qf)
q_pids = np.asarray(np.concatenate(q_pids))
q_camids = np.asarray(np.concatenate(q_camids))

其中q_pids包含的是numpy数组的列表,通过np.concatenate()可以直接转化成为列数组。如果先前不用pids = pids.numpy().astype(np.int16)来将pids转化为numpy数组,用torch.cat()也可以达到同样的效果。

评论

此博客中的热门博文

使用ssh反向代理+shadowsocks实现内网穿透

shadowsocks中转

ubuntu 16.04 reboot命令慢的原因