open3d.core.Tensor.argmin函数用于在指定的维度上返回张量中最小值的下标。
argmin(dim=None, keepdim=False)
import open3d.core as o3c
import numpy as np
# 创建一个张量
t1 = o3c.Tensor(np.array([[2, 4, 6],
[8, 3, 9],
[1, 5, 7]]).astype(np.float32))
# 在整个张量上查找最小值下标
res1 = t1.argmin()
# 在每一列上查找最小值下标
res2 = t1.argmin(dim=0)
# 在每一行上查找最小值下标
res3 = t1.argmin(dim=1)
print(res1)
print(res2)
print(res3)
输出结果如下所示：
[6]
[2 1 0]
[0 1 0]