max
方法可用于计算 Tensor
的最大值。该方法返回一个 1
维张量,其值为输入张量在指定维度上的最大值。
def max(self, dim: int = -1, keepdim: bool = False) -> Tensor
dim
:指定在哪个维度计算最大值,可以是一个0维到n维的数值。默认为最后一维 -1
。取值范围为 [-dim, dim-1]
keepdim
:是否将结果保留在一个长度等于1
的维度上。默认为 False
。返回一个 1
维张量,其值为输入张量在指定维度上的最大值。
import open3d.core as o3c
import torch
# 创建一个2维张量
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])
# 将torch张量转为Open3D张量
o3tensor = o3c.Tensor(tensor)
# 在最后一维上计算最大值
max_value = o3tensor.max()
# 打印结果
print(max_value)
结果为:
[3]
import open3d.core as o3c
tensor = o3c.ones((2, 3, 4))
max_value = tensor.max(dim=2, keepdim=True)
print(max_value)
结果为:
[1. 1. 1. 1.],
[1. 1. 1. 1.],
dtype=float32
size=(2, 3, 1)
IndexError
:如果指定的维度 dim
超出了张量的范围。