torch.booltensor
时间: 2023-09-07 21:18:11 浏览: 70
PyTorch的torch.cat用法
5星 · 资源好评率100%
torch.booltensor是用于创建一个布尔类型的张量的函数。它接受一个可迭代对象作为输入,并将其转换为布尔类型的张量。例如,你可以使用以下代码创建一个包含布尔值的张量:
```
import torch
data = [True, False, True, True]
bool_tensor = torch.BoolTensor(data)
print(bool_tensor)
```
输出将是:
```
tensor([ True, False, True, True])
```
注意:在最新的PyTorch版本中,`torch.BoolTensor()`已经被弃用,推荐使用`torch.tensor()`并指定dtype参数为`torch.bool`来创建布尔类型的张量。
阅读全文