(六)Tensor维度变换

2020-02-12  本文已影响0人  zelda2333

操作

1.View/reshape

image.png

2.Squeeze/unsqueeze

unsqueeze(position)
position范围:[-a.dim()-1, a.dim()+1)




example
example

squeeze


image.png

3.Transpose/t/permute

permute

4. Expand/repeat

Expand
Repeat
上一篇 下一篇

猜你喜欢

热点阅读