PyTorch torch.any 函数

PyTorch torch.any 函数 Pytorch torch 参考手册 torch.any 是 PyTorch 中用于判断是否有元素为 True 的函数。函数定义 torch.any(input, dim, keepdim, out) 使用示例实例 [mycode4 type='python'] import torch x = torch.tensor([[True, False,..

Updated 7/16/2026

Tutorial detail pages default to noindex.

PyTorch torch.any 函数

Pytorch torch 参考手册

torch.any 是 PyTorch 中用于判断是否有元素为 True 的函数。

函数定义

torch.any(input, dim, keepdim, out)

使用示例

实例

import torch

x = torch.tensor([[True, False, False], [False, False, False]])

print("有 True:", torch.any(x))
print("dim=0 有 True:", torch.any(x, dim=0))

输出结果为：

有 True: tensor(True)
dim=0 有 True: tensor([True, False, False])

Pytorch torch 参考手册