ARTICLE DETAIL

资讯详情

深耕网站建设、视觉设计与SEO优化的一线实战洞察。

pytorch进行转换的一些函数

pytorch进行转换的一些函数

https://zhuanlan.zhihu.com/p/583942614

pytorch tensor 操作:合并、分割、维度变换

 

其中expand有些补充,expand只能扩1的维度,或者在最前面维度新扩

repeat针对各个维度拷贝几倍

 

view和reshape的区别,可以参考这个:

https://zhuanlan.zhihu.com/p/1919706656255750256

如果transpose的话,会破坏 is_contiguous,就不能view,只能reshape 

 

返回列表