分类：判断索引

2021
04-25

Pytorch 使用tensor特定条件判断索引

torch.where()用于将两个broadcastable的tensor组合成新的tensor，类似于c++中的三元操作符“?:”区别于pythonnumpy中的where()直接可以找到特定条件元素的index想要实现numpy中where()的功能，可以借助nonzero()对应numpy中的where()操作效果:补充：Pytorchtorch.Tensor.detach()方法的用法及修改指定模块权重的方法detachdetach的中文意思是分离，官方解释是返回一个新的Tensor,从当前的计算图中分离出来需要注意的是，返... 继续阅读 >