【PyTorch】PyTorch中张量(Tensor)切片操作

PyTorch深度学习总结

第三章 PyTorch中张量(Tensor)切片操作

---

---

一、前言

上文介绍了PyTorch中改变张量(Tensor)形状的操作，本文主要介绍张量切片操作。

---

二、获取张量中的元素

1、切片（行、列数）方法

①

# 引入库
import torch

# 生成张量
A = torch.arange(9).reshape(3, 3)
print(A)

生成张量A：
tensor(
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]])

---

现截取A[0]:

print(A[0]) # 截取最外围括号内第一个括号的内容,第一个维度第一行的内容

结果为：
tensor([0, 1, 2])

②

# 引入库
import torch

# 生成张量
B = torch.arange(