Pytorch基础：Tensor索引会导致创建新张量

相关阅读

Pytorch基础https://blog.csdn.net/weixin_45791458/category_12457644.html?spm=1001.2014.3001.5482

---

        本文来源于博主无意之中的一个发现，当对tensor进行相同的索引操作时，每次返回的tensor对象的id值不同；对于一般的列表，每次返回的对象id值应该是相同（即意味着每次都是返回同一个对象）。

import torch# 创建普通的列表
y = [1, 2, 3, 4]y_1 = y[0]
y_2 = y[0]print(id(y_1) == id(y_2))
true# 创建一个张量
x = torch.rand(3, 3, 3)x_1 = x[0]
x_2 = x[0]print(id(x_1) == id(x_2))
false
print(x_1.storage().data_ptr()==x_2.storage().data_ptr()==x.storage().data_ptr())
true

        索引的结果与原张量共享底层存储，但是由一个新的张量表示，且每次索引都会创建新的张量，需要注意此处张量与列表的区别，对于列表来说，无论索引几次，得到的结果都是同一个对象。