心如止水_c736

关注

tf.squeeze

心如止水_c736

关注

阅读 152

2023-01-16


从张量的形状中删除大小为 1 的维度

import tensorflow as tf

a=tf.constant([[[1,2,3,1]]])

print(a.shape)

a1=tf.squeeze(a)

print(a1.shape)

(1, 1, 4)
(4,)

​​文档链接​​



相关推荐

何晓杰Dev

【SENet】Squeeze-and-Excitation Networks

何晓杰Dev 60 0 0

霍华德

pytorch中squeeze()和unsqueeze()函数

霍华德 46 0 0

犹大之窗

Pytorch函数view、permute、squeeze、usqueeze

犹大之窗 72 0 0

醉东枫

Can not squeeze dim[1], expected a dimension of 1

醉东枫 94 0 0

践行数据分析

tensorflow中squeeze与expend_dims

践行数据分析 26 0 0

楚木巽

Pytorch:cat、stack、squeeze、unsqueeze的用法

楚木巽 21 0 0

滚过红尘说红尘

python tf

滚过红尘说红尘 83 0 0

舍予兄

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

舍予兄 171 0 0

M4Y

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

M4Y 79 0 0

杰克逊爱学习

tensorflow语法【zip、tf.tile、tf.truncated_normal、tf.data.Dataset.from

杰克逊爱学习 163 0 0

精彩评论(0)

0 0 举报