pytorch-RefineDet------PriorBox
相关函数 1、torch.clamp() torch.clamp(input, min, max, out=None) --> Tensor 将输入input Tensor中每个元素约束到【min,max】,返回一个截断后的new Tensor 例子: input = torch.Tensor([2,1,0.5,-0.5]) output = torch.clamp(input,0,1) print(output) ''' output Out[16]: tensor([1.000, 1.000, 0.500, 0.000]) ''' 2、itertools.product( ) product(*iterables,repeat=1) --> product object iterable是可迭代对象,repeat指定iterable重复几次,即product(A,repeat=3)等价于product(A,A,A) 该函数用于求输入迭代对象的笛卡尔集,相当于一个前天的for循环。product返回的是一个迭代器。 import itertools.product as product list1=['a','b'] list2=['c','d'] for item in product(list1,list2): print(item) ''' ... ('a','c') (