改变Tensor尺寸的操作 1.tensor.view tensor.view方法,可以调整tensor的形状,但必须保证调整前后元素总数一致。view不会改变自身数据,返回的新的tensor与源tensor共享内存,即更改其中一个,另外一个也会跟着改变。 例: In: import torch as t a = t.ar
改变Tensor尺寸的操作
2.tensor.unsqueeze 和 tensor.squeeze tensor.unsqueeze 和 tensor.squeeze分别用于增加或减少tensor的某一维度。 例:
3.tensor.resize tensor.resize是另外一种可以调整tensor尺寸的方法,但与view不同,它可以修改tensor的尺寸。如果新尺寸超过了原尺寸,会自动分配新的内存空间;如果新尺寸小于原尺寸,则之前的数据依旧会保存 例:
|
2019-06-18
2019-07-04
2021-05-23
2021-05-27
2021-05-27