2022 iThome 鐵人賽
分享至
因為文本序列化需要將單詞對應的數字存入字典,所以我們現在要先將每個單詞獨立出來用dataset直接區會有問題因為如果input進dataset的資料是字符串型,dataset會將它自動zip,所以我們要用自定義一個collate_fn方法來將他整理來解決異常
這是我們數據原來的模樣
這是實現collate_fn的程式
IT邦幫忙