pytorch中用索引操作原张量不变
在《动手深度学习-Pytorch》教程中看到了如下内容:
索引
我们还可以使用类似NumPy的索引操作来访问Tensor的一部分,需要注意的是:索引出来的结果与原数据共享内存,也即修改一个,另一个会跟着修改。
y = x[0, :]
y += 1
print(y)
print(x[0, :]) # 源tensor也被改了
输出
tensor([1.6035, 1.8110, 0.9549])
t...
原创
2020-01-09 21:12:49 ·
421 阅读 ·
0 评论