libtorch 数组索引和张量操作 与pytorch比较
摘要:
数组索引 libtorch torchTensor idx=torchzeros({270,360}); torchTensor t=torchzeros({255}); //idx里的值作为t的索引 torch::Tensor out=t.index({idx}) python out=t[idx 阅读全文
posted @ 2021-09-01 09:39 逆风的泰迪man 阅读(299) 评论(0) 推荐(0)
浙公网安备 33010602011771号