luwanglin

摘要： pytorch常用函数总结 torch.max(input,dim) 求取指定维度上的最大值，,返回输入张量给定维度上每行的最大值，并同时返回每个最大值的位置索引。比如： demo.shape Out[7]: torch.Size([10, 3, 10, 10]) torch.max(demo,1) 阅读全文

摘要： #numpy函数笔记 np.isin用法 np.isin(a,b) 用于判定a中的元素在b中是否出现过，如果出现过返回True,否则返回False,最终结果为一个形状和a一模一样的数组。（注意：这里的a和b是像数组类型就行，比如列表的话，传入进去之后，numpy会自动将其转化为numpy数组） 但是 阅读全文