zae

摘要： 函数作用 torch.where()函数可以是按照给定的条件合并两个tensor； 条件满足的位置取前者，否则取后者。 代码示例 >>> import torch >>> a=torch.randn(3,5) >>> b=torch.ones(3,5) >>> a tensor([[-0.0310, 阅读全文