给定输入如下:
tensor([[[1.9392, -1.9266, 0.9664],
[0.0000, -1.9266, 0.9664],
[0.0000, -0.0000, 0.9664]]])
Run Code Online (Sandbox Code Playgroud)
我想要的输出是:
tensor([[[0.4596, 0.0096, 0.1737],
[0.0000, 0.0096, 0.1737],
[0.0000, -0.0000, 0.1737]]])
Run Code Online (Sandbox Code Playgroud)
即仅计算上三角元素的函数。