0
点赞
收藏
分享

微信扫一扫

torch的any()函数

王老师说 2022-08-20 阅读 100


作用

这个函数用于判断是否有true项。具体示例如下:

torch的any()函数_深度学习

报错

其实很不理解有时候代码写着看着没有问题,但是为啥就是报错?
比如我今天写了下面一段代码:

if torch.any(gold):
total_golden+=1

就一直报错:​​RuntimeError: all only supports torch.uint8 and torch.bool dtypes​​ 关键是我的gold的的确确是tensor。


举报

相关推荐

0 条评论