践行数据分析

关注

tensorflow中squeeze与expend_dims

践行数据分析

关注

阅读 26

2022-01-08

tf.squeeze

  • z = tf.squeeze(y) #删除y中所有为1的维度;
  • z = tf.squeeze(y,[3,5]) #删除[3,5]中所有为1的维度;

tf.expand_dims

  • tf中没有tf.unsqueeze,一般写tf.expand_dims()扩增维度;
  • one_img = tf.expand_dims(one_img, 0) #第0维插入1维
  • one_img = tf.expand_dims(one_img, -1) #-1表示最后一维

相关推荐

霍华德

pytorch中squeeze()和unsqueeze()函数

霍华德 46 0 0

舍予兄

pytorch中squeeze()和unsqueeze()函数介绍

舍予兄 171 0 0

酷子腿长一米八

torch中 squeeze 和 unsqueeze 的不同用法

酷子腿长一米八 38 0 0

M4Y

Pytorch维度操作-unsqueeze、squeeze、view与permute

M4Y 79 0 0

烟中雯城

关于torch.unsqueeze与torch.squeeze

烟中雯城 115 0 0

深夜瞎琢磨

pytorch—Tensorflow中模块与常见问题

深夜瞎琢磨 47 0 0

静守幸福

numpy 中降维函数ravel()、flatten()、squeeze()、reshape()笔记

静守幸福 165 0 0

phpworkerman

PyTorch中torch.squeeze() 和torch.unsqueeze()用法

phpworkerman 55 0 0

kmoon_b426

pytorch中的torch.unsqueeze和squeeze张量维度变化问题

kmoon_b426 232 0 0

产品喵dandan米娜

tensorflow中slim详解

产品喵dandan米娜 139 0 0

精彩评论(0)

0 0 举报