torch中关于torch.max()和torch.min()函数的理解
简介在tensor类型的数据中,max和min函数常用来比较两个tensor数据的大小,或者取出tensor数据中的最大值。关于max函数和min函数的用法有以下几种场景:对于tensorA和tensorB:torch.max(tensorA):返回tensor中的最大值。 torch.mac(tensorA,dim):dim表示指定的维度,返回指定维度的最大数和对应下标 torch...
原创
2019-07-12 13:42:36 ·
19171 阅读 ·
1 评论