1:where(condition,x,y)
[注]返回一个和a,b shape相同的tensor,返回的tensor内容根据选择条件condition从a,b中进行选择。
如上图:选择条件为cond>0.5会返回一个元素类型为torch.unit8类型的0 or 1的tensor。0代表false,1代表true。
2024-01-08 16:25:10
1:where(condition,x,y)
[注]返回一个和a,b shape相同的tensor,返回的tensor内容根据选择条件condition从a,b中进行选择。
如上图:选择条件为cond>0.5会返回一个元素类型为torch.unit8类型的0 or 1的tensor。0代表false,1代表true。