自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(2)
  • 收藏
  • 关注

原创 Pytorch最大值最小值函数:torch.max()、torch.min()

若dim=0,keepdim=True,将返回每列上的最大值,与对应行的索引,值与索引的格式保留了列的维度信息。keepdim=True时,返回的值与索引的shape为[1, 3],列的维度信息被保留下来。当传入dim时,将返回dim对应维度上的最大值与对应的dim维度的索引。若input为一个二维的Tensor,那么dim=0时将在列上寻找最大值,并返回对应行的索引;dim=1时将在行上寻找最大值,并返回对应列的索引。当只传入input时,则在整个input中寻找最大值,只返回最大值,不返回索引。

2023-08-12 23:02:41 2326 1

原创 Pytorch中比较大小的相关函数:torch.gt()、torch.ge、torch.lt()、torch.le()、torch.eq()、torch.ne()

这里只以torch.gt()函数举例,其他函数用法与torch.gt()函数类似。Pytorch官方介绍:torch.gt()函数原型:返回值为input与other大小比较的结果。若input大于other返回True,否则返回False。如果input与other的shape相同,那么则在对应位置进行比较;如果input与other的shape不同,则会用到广播机制。当input与other的shape相同时,直接对应位置的值进行比较。

2023-08-12 21:53:39 1084

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除