pytorch中bool类型是可以作为索引的,同时也支持sum等操作。
索引时相当于取出为True等位置的元素之后组成一个新的tensor.
pytorch中bool类型的张量作为索引
最新推荐文章于 2023-06-22 00:05:33 发布
pytorch中bool类型是可以作为索引的,同时也支持sum等操作。
索引时相当于取出为True等位置的元素之后组成一个新的tensor.