tensor类型的元素怎么填充对角线,其余元素置为0
时间: 2024-03-08 12:49:44 浏览: 140
PyTorch中Tensor的数据类型和运算的使用
可以使用 PyTorch 中的函数 `torch.diag()` 来实现。具体的做法如下:
```python
import torch
# 定义一个一维张量
x = torch.tensor([1, 2, 3])
# 构造对角线张量
diag = torch.diag(x)
print(diag)
# 输出结果为:
# tensor([[1, 0, 0],
# [0, 2, 0],
# [0, 0, 3]])
```
上述代码中,首先使用 `torch.diag()` 函数将一维张量 `x` 转化为对角线张量 `diag`,其他位置上的元素为 0。
阅读全文