data_ptr方法是tensor向量的一个很重要的方法,它的作用是返回tensor首元素的内存地址,常用来判断两个tensor是不是共享内存,如下所示:
data_t = data.t() # Transpose
print(data.data_ptr() == data_t.data_ptr()) # True
因为Transpose操作只是改变了stride信息,data_t 和data指向同一个memory。
data_ptr方法是tensor向量的一个很重要的方法,它的作用是返回tensor首元素的内存地址,常用来判断两个tensor是不是共享内存,如下所示:
data_t = data.t() # Transpose
print(data.data_ptr() == data_t.data_ptr()) # True
因为Transpose操作只是改变了stride信息,data_t 和data指向同一个memory。