上面代码中的0:dataset.shape[1]什么意思
时间: 2024-05-17 16:15:52 浏览: 118
浅谈tensorflow中dataset.shuffle和dataset.batch dataset.repeat注意点
这里的dataset是一个二维数组,shape是numpy数组的属性,表示数组的形状,返回一个元组,第一个元素表示数组的行数,第二个元素表示数组的列数。因此,dataset.shape[1]表示数组dataset的列数,即第二维的长度。在代码中,0:dataset.shape[1]表示从第0列到第(dataset.shape[1]-1)列的所有列。
阅读全文