我用torch::gt函数得到了大于阈值的index,但是这个index在libtorch里怎么用呢?没办法像python一样使用[:][:][[index]的用法啊。
-
在libtorch里,怎么根据阈值选择特定的行?
-
刘看山 |
这个就是用where,
类似于python里面的操作,类似于numpy。
我们社区有一些libtorch的项目,里面的操作比如根据概率选择候选框就有示例程序。
在libtorch里,怎么根据阈值选择特定的行?
我用torch::gt函数得到了大于阈值的index,但是这个index在libtorch里怎么用呢?没办法像python一样使用[:][:][[index]的用法啊。
这个就是用where,
类似于python里面的操作,类似于numpy。
我们社区有一些libtorch的项目,里面的操作比如根据概率选择候选框就有示例程序。