torch
关注
分享
扫一扫
Frank（Zhiyang-Dou）
A Ph.D. candidate at HKU. Mainly focus on CG.
展开
-
pytorch view与reshape区别
区别存储view:返回具有新形状的张量。返回的张量将与原始张量共享基础数据。reshape: torch.reshape可能返回原始张量的副本或视图->不确定的要求reshape（）可以在连续和非连续的张量上运行，而view（）只能在连续的张量上运行。...原创 2020-11-04 14:34:55 · 1201 阅读 · 0 评论 -
增加tensor维度 torch
综述增加tensor的维度.使用unsqueeze在指定维度上增加。代码给定一个tensor假设其维度为：print(a.shape)torch.Size([2, 2048])b = a.unsqueeze(1)print(a.shape)torch.Size([2, 1,2048])原创 2020-11-04 09:21:50 · 7298 阅读 · 0 评论