pytorch tensor.all() tensor.any()

最新推荐文章于 2023-10-31 11:47:03 发布

littlehaes

最新推荐文章于 2023-10-31 11:47:03 发布

阅读量1.9w

点赞数 26

分类专栏： Pytorch

本文链接：https://blog.csdn.net/littlehaes/article/details/103945686

版权

Pytorch 专栏收录该内容

12 篇文章 4 订阅

订阅专栏

welcome to my blog

`tensor.any()`功能: 如果张量tensor中存在一个元素为True, 那么返回True; 只有所有元素都是False时才返回False

import torch
a = torch.tensor([True,True,False])
print(a.any())
# 打印结果 tensor(True)

b = torch.tensor([False, False, False])
print(b.any())
# 打印结果 tensor(False)

`tensor.all()`功能: 如果张量tensor中所有元素都是True, 才返回True; 否则返回False

import torch
a = torch.tensor([True,True,False])
print(a.all())
# 打印结果 tensor(False)

b = torch.tensor([True,True,True])
print(b.all())
# 打印结果 tensor(True)