【python】pytorch中如何使用DataLoader對數據集進行批處理

第一步: 我們要創建torch能夠識別的數據集類型(pytorch中也有很多現成的數據集類型,以後再說)。 首先我們建立兩個向量X和Y,一個作爲輸入的數據,一個作爲正確的結果:      隨後我們需要把X和Y組成一個完整的數據集,並轉化爲pytorch能識別的數據集類型:      我們來看一下這些數據的數據類型:       可以看出我們把X和Y通過Data.TensorDataset() 這個
相關文章
相關標籤/搜索