tf.expand_dims()
tf.tf.expand_dims(input, axis=None, name=None, dim=None)
參數介紹:
- input:欲升維的tensor。
- axis:插入新維度的索引位置,維度索引值得軸從零開始,指定軸是負數,則從最後向後進入進行計數。
- name:輸出tensor名稱。
- dim:一般不會指定。
tf.squeeze()
td.squeeze(input, squeeze_dims=None, name=None)
- input:降低維度的張量。
- squeeze_dims:list[int]類型,表示需刪除的維度索引。默認為[],即刪除大小為1的維度。