Pytorch: ブールテンソルを作成します (型: torch.ByteTensor)? 質問する

Pytorch: ブールテンソルを作成します (型: torch.ByteTensor)? 質問する

ブール値のみを含むテンソルを作成したい。Matlabでは次のようになる。

a = false(10,1)

ベストアンサー1

これはもっと経済的ではないでしょうか(ただし時間はかかります):

a = torch.zeros(10, dtype=torch.bool)

または、古いバージョンでは、

a = torch.zeros(10, dtype=torch.uint8)

(bool へのポインタを提供してくれた @drevicko に感謝します。)

おすすめ記事