Check All Values from One Tensor Equal to One Value
作者:XD / 发表: 2023年11月16日 06:26 / 更新: 2023年11月16日 06:26 / 编程笔记 / 阅读量:670
Check All Values from One Tensor Equal to One Value
import torch
tensor = torch.tensor([[[8, 8, 8, ..., 8, 8, 8]],
[[8, 8, 8, ..., 8, 8, 8]],
[[8, 8, 8, ..., 8, 8, 8]]], dtype=torch.int8)
is_all_eight = torch.all(tensor == 8)
print(is_all_eight) # True or False
相关标签