一、where
1）torch.where(condition, x, y) # condition是条件，满足条件就返回x，不满足就返回y
2）特点，相比for循环的优点是：可以布置在GPU上运行
二、gather
1）官方解释：根据指定的维度和索引值来筛选值
2）举例
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
猜您在找
Pytorch张量高阶操作
tensorflow添加新操作（Op）
tf.gather()、tf.gather_nd()、tf.batch_gather()、tf.where()和tf.slice()
理解pytorch几个高级选择函数（如gather）
Pytorch并行计算：nn.parallel.replicate, scatter, gather, parallel_apply
PyTorch分布式训练详解教程 scatter, gather & isend, irecv & all_reduce & DDP
Pytorch：Tensor 张量操作
Pytorch 剪枝操作实现
Pytorch 基本操作
pytorch-Flatten操作