diag
函数用于对给定的输入张量进行对角线扩展。从对角线上取出元素,按照给定的维度进行重组。
def diag(input: open3d.core.Tensor, diagonal: int = 0) -> open3d.core.Tensor:
pass
import open3d.core as o3c
import numpy as np
a = np.array([[1, 2],
[3, 4],
[5, 6]])
t = o3c.Tensor(a)
print(o3c.tensor_diag(t))
输出结果:
[[1 0 0]
[0 4 0]
[0 0 6]]