CudnnRNNCanonicalToParamsV2

แปลงพารามิเตอร์ CudnnRNN จากรูปแบบมาตรฐานเป็นรูปแบบที่ใช้งานได้ รองรับการฉายภาพใน LSTM

เขียนชุดตุ้มน้ำหนักลงในบัฟเฟอร์พารามิเตอร์ทึบแสง เพื่อให้สามารถนำไปใช้ในการฝึกอบรมหรือการอนุมานที่กำลังจะมีขึ้น

โปรดทราบว่าบัฟเฟอร์พารามิเตอร์อาจเข้ากันไม่ได้กับ GPU ที่แตกต่างกัน ดังนั้นการบันทึกและการฟื้นฟูใดๆ ควรถูกแปลงเป็นและจากน้ำหนักและอคติตามรูปแบบบัญญัติ

num_layers: ระบุจำนวนเลเยอร์ในโมเดล RNN num_units: ระบุขนาดของสถานะที่ซ่อนอยู่ input_size: ระบุขนาดของสถานะอินพุต ตุ้มน้ำหนัก: รูปแบบมาตรฐานของตุ้มน้ำหนักที่สามารถใช้เพื่อบันทึกและฟื้นฟู มีแนวโน้มที่จะเข้ากันได้มากกว่าคนรุ่นต่างๆ อคติ: รูปแบบบัญญัติของอคติที่สามารถนำมาใช้ในการบันทึกและฟื้นฟู มีแนวโน้มที่จะเข้ากันได้มากกว่าคนรุ่นต่างๆ num_params_weights: จำนวนเมทริกซ์พารามิเตอร์น้ำหนักสำหรับทุกเลเยอร์ num_params_biases: จำนวนเวกเตอร์พารามิเตอร์อคติสำหรับทุกเลเยอร์ rnn_mode: ระบุประเภทของโมเดล RNN input_mode: ระบุว่ามีการฉายภาพเชิงเส้นระหว่างอินพุตและการคำนวณจริงก่อนเลเยอร์แรกหรือไม่ อนุญาตให้ใช้ 'skip_input' เมื่อ input_size == num_units เท่านั้น 'auto_select' หมายถึง 'skip_input' เมื่อ input_size == num_units; มิฉะนั้นจะหมายถึง 'linear_input' ทิศทาง: ระบุว่าจะใช้โมเดลแบบสองทิศทางหรือไม่ dir = (ทิศทาง == แบบสองทิศทาง) ? 2 : 1 การออกกลางคัน: ความน่าจะเป็นของการออกกลางคัน เมื่อตั้งค่าเป็น 0 การออกกลางคันจะถูกปิดใช้งาน เมล็ด: ส่วนที่ 1 ของเมล็ดเพื่อเริ่มต้นการออกกลางคัน seed2: ส่วนที่ 2 ของเมล็ดเพื่อเริ่มต้นการออกกลางคัน num_proj: มิติเอาท์พุตสำหรับเมทริกซ์การฉายภาพ หากไม่มีหรือ 0 จะไม่มีการฉายภาพ