2021-04-27
Tensor的简单的数据操作
import torch
x = torch.rand(5, 3)
# 1 用view改变tensor([])的形状:新源tensor共享内存,view仅仅改变了对此张量的观察角度
y = x.view(15)
z = x.view(-1, 5) # -1所指的维度可以根据其他维度的值推出来
print(x.size(), y.size(), z.size())
# 2 用reshape改变形状,返回一个新的tensor副本
Z = x.reshape()
print(Z
原创
2021-04-27 16:12:29 ·
117 阅读 ·
2 评论