pytorch常用函数view、sum、sequeeze、cat和chunk
view()函数
view()
相当于reshape、resize
,重新调整Tensor
的形状。
- 指定调整形状之后的维度
import torch
re = torch.tensor([1, 2, 3, 4, 5
原文地址:https://blog.csdn.net/weixin_46530492/article/details/142714050
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!