- 博客(2)
- 收藏
- 关注
RSS订阅原创 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关注的人