我想创建一个仅包含布尔值的张量。在Matlab中
a = false(10,1)
Run Code Online (Sandbox Code Playgroud)
已经找到了:
a = torch.zeros(10)
b = a.type(torch.ByteTensor)
Run Code Online (Sandbox Code Playgroud)
小智 5
这不是更经济(虽然更长):
a = torch.zeros(10, dtype=torch.uint8)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
9383 次 |
| 最近记录: |