1. 程式人生 > 實用技巧 >TensorFlow2.0學習筆記-----索引與切片

TensorFlow2.0學習筆記-----索引與切片

1. 索引

a = tf.ones([1,5,5,3]) 建立一個shape為[1,5,5,3]的全1張量,可以理解為一張圖片,大小5x5,有三個通道

a[1,2]和a[1][2]是同一種索引方式的不同表達形式,都可以

a[1,2]輸入numpy風格的索引

2.切片

:前後分別是start和end

;代表該維度的所有值都取