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()收集:挺难的不太懂

 

 

 

posted @ 2021-07-11 17:41  收购阿里巴巴  阅读(34)  评论(0编辑  收藏  举报