triu()
是Open3D中的一个函数,可以从给定张量的主对角线及其以下部分中选择元素并返回新张量,其余元素全部为0。
open3d.core.Tensor.triu(k: int = 0, out=None) -> open3d.core.Tensor
k
:选取从主对角线往上的k
行。out
:输出张量。返回一个新张量。
import open3d.core as o3c
import numpy as np
tensor = o3c.Tensor(
np.array([
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
])
)
triu_tensor = tensor.triu()
print(triu_tensor)
输出结果:
[[1, 2, 3],
[0, 5, 6],
[0, 0, 9]]