创建一个值为[1,2,3,4,5]的一维张量,切片得到一个值为[2,4]的张量。
时间: 2024-04-07 17:12:29 浏览: 56
tensorflow实现对张量数据的切片操作方式
好的,这个问题很简单,可以用以下代码来创建和切片张量:
```python
import torch
# 创建一维张量
x = torch.tensor([1, 2, 3, 4, 5])
# 切片得到目标张量
y = x[1:4:2]
# 打印目标张量
print(y)
```
输出结果为:`tensor([2, 4])`
希望能够帮到你!
阅读全文