7:高阶张量操作

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。

 2:gather()收集:挺难的不太懂

原文地址:https://www.cnblogs.com/jiafeng1996/p/14999089.html