企业如何推广网站,西乡网站建设,WordPress黑镜主题,品牌展示型网站源码torch.max 和 torch.argmax 的区别
1.torch.max
torch.max(input, dim, maxNone, max_indicesNone, keepdimFalse) -- (Tensor, LongTensor)
作用#xff1a;找出给定tensor的指定维度dim上的上的最大值#xff0c;并返回最大值在该维度上的值和位置索引。 应用举…torch.max 和 torch.argmax 的区别
1.torch.max
torch.max(input, dim, maxNone, max_indicesNone, keepdimFalse) -- (Tensor, LongTensor)
作用找出给定tensor的指定维度dim上的上的最大值并返回最大值在该维度上的值和位置索引。 应用举例 例1——返回相应维度上的最大值,并返回最大值的位置索引
a torch.randn(4, 4)
a
tensor([[-1.2360, -0.2942, -0.1222, 0.8475],[ 1.1949, -1.1127, -2.2379, -0.6702],[ 1.5717, -0.9207, 0.1297, -1.8768],[-0.6172, 1.0036, -0.6060, -0.2432]])
torch.max(a, 1)
torch.return_types.max(valuestensor([0.8475, 1.1949, 1.5717, 1.0036]),
indicestensor([3, 0, 0, 1]))例2——如果max的参数只有一个tensor则返回该tensor里所有值中的最大值。
a torch.randn(4, 4)
a
tensor([[ 0.4997, 0.8054, 0.1761, 0.3055],[-1.2234, 0.3823, 0.2266, -2.9062],[ 0.4390, -1.0142, -0.5314, -1.7095],[-0.2296, -0.4230, -0.7446, -0.0828]])
torch.max(a)
tensor(0.8054)
例3——如果max的参数是两个相同shape的tensor则返回两tensor元素对应位置的最大值的新tensor
a torch.randint(2, 10,(6,4))
a
tensor([[8, 7, 3, 5],[2, 8, 3, 4],[3, 2, 5, 5],[4, 7, 5, 2],[2, 9, 3, 8],[4, 4, 2, 2]])
b torch.randint(2, 10,(6,4))
b
tensor([[9, 8, 9, 2],[4, 3, 3, 4],[6, 9, 2, 7],[4, 3, 2, 7],[4, 4, 9, 2],[8, 2, 6, 2]])
torch.max(a, b)
tensor([[9, 8, 9, 5],[4, 8, 3, 4],[6, 9, 5, 7],[4, 7, 5, 7],[4, 9, 9, 8],[8, 4, 6, 2]])2. torch.argmax
函数定义 torch.argmax(input, dim, keepdimFalse) → LongTensor 作用返回输入张量中指定维度的最大值的索引。 应用举例 例1——指定维度返回相应维度最大值的索引
a torch.randn(4, 4)
a
tensor([[ 1.3398, 0.2663, -0.2686, 0.2450],[-0.7401, -0.8805, -0.3402, -1.1936],[ 0.4907, -1.3948, -1.0691, -0.3132],[-1.6092, 0.5419, -0.2993, 0.3195]])
torch.argmax(a, dim1)
tensor([ 0, 2, 0, 1])例2——不指定维度返回整体上最大值的序号
a torch.randint(9,(3, 3))
a
tensor([[5, 2, 2],[7, 2, 0],[8, 0, 6]])
torch.argmax(a)
tensor(6)3.torch.min
用法同max
4.torch.argmin
用法同argmax
