摘要:
*args:是可变参数(不知道型参里有多少个参数),将接受的参数自动转化为列表(list) 如下程序: def func(*arge): sum=0 for i in arge: sum=sum+i*i return sum 当调用func函数时,不必将函数的型参写成list的格式: func(1, 阅读全文
posted @ 2020-08-28 13:53
SnailWorks
阅读(303)
评论(0)
推荐(0)
摘要:
在进行分类问题时,通常需要使用torch.max()函数,来求预测值的索引 output=torch.max(input,dim) input:是出入的数据 dim:数据的维度,对于二维图像来说,通常dim为0和1,当dim=0时,进行进行每列的最大数值索引,当dim=1时,进行每行的最大值索引, 阅读全文
posted @ 2020-08-28 09:25
SnailWorks
阅读(392)
评论(0)
推荐(0)
浙公网安备 33010602011771号