自学内容网 自学内容网

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)!