sequeeze(dim):用于在指定位置添加一个大小为1的新维度
例如:
其他的大家自行尝试!
squeeze():squeeze函数用于去除张量中所有大小为1的维度,注意当没有传入任何参数的时候,去掉的是所有大小为1的维度。
例如:
sequeeze(dim):用于在指定位置添加一个大小为1的新维度
例如:
其他的大家自行尝试!
squeeze():squeeze函数用于去除张量中所有大小为1的维度,注意当没有传入任何参数的时候,去掉的是所有大小为1的维度。
例如: