torch.all¶
- torch.all(input) Tensor ¶
测试
input
中的所有元素是否都计算为 True。注意
此函数与 NumPy 的行为匹配,对于除 uint8 之外的所有支持的数据类型,返回数据类型为 bool 的输出。对于 uint8,输出的数据类型本身就是 uint8。
示例
>>> a = torch.rand(1, 2).bool() >>> a tensor([[False, True]], dtype=torch.bool) >>> torch.all(a) tensor(False, dtype=torch.bool) >>> a = torch.arange(0, 3) >>> a tensor([0, 1, 2]) >>> torch.all(a) tensor(False)
- torch.all(input, dim, keepdim=False, *, out=None) Tensor
对于给定维度
dim
中input
的每一行,如果该行中的所有元素都计算为 True,则返回 True,否则返回 False。如果
keepdim
为True
,则输出张量与input
大小相同,除了在维度dim
上大小为 1。否则,dim
将被压缩(参见torch.squeeze()
),导致输出张量减少 1(或len(dim)
)个维度。- 参数
- 关键字参数
out (Tensor, optional) – 输出张量。
示例
>>> a = torch.rand(4, 2).bool() >>> a tensor([[True, True], [True, False], [True, True], [True, True]], dtype=torch.bool) >>> torch.all(a, dim=1) tensor([ True, False, True, True], dtype=torch.bool) >>> torch.all(a, dim=0) tensor([ True, False], dtype=torch.bool)