DargonWain的博客

https://stackoverflow.com/questions/63479765/whats-the-best-way-of-checking-whether-a-torchtensor-is-emptyif (!tensor.numel()){ std::cout<<"tensor is empty!" << std::endl; // do other checks you wish to do}

文章目录方法一 利用Numpy 广播机制方法二 通过将RGB一维数组对比转换为标量对比方法三 通过hash加矩阵运算加速对比(耗内存较大)测试:参考:在做图像语义分割时常常会遇到标注是像素颜色值类型,这时候我们需要将图像转换为单通道图像再针对不同的颜色值赋相应的类别,一个简单的思路就是对图像进行逐像素遍历,但python下对多维数组逐个进行for循环是很耗时的一件事.numpy底层对矩阵运算做了很多的优化,这时候擅用numpy本身自带的运算可以极大帮助我们加速这一过程.方法一 利用Numpy 广播机制