pytorch中bool类型是可以作为索引的,同时也支持sum等操作。
索引时相当于取出为True等位置的元素之后组成一个新的tensor.
pytorch中bool类型的张量作为索引
猜你喜欢
转载自blog.csdn.net/mch2869253130/article/details/118088278
今日推荐
周排行