คำนวณขนาดของน้ำหนักที่โมเดล Cudnn RNN สามารถใช้ได้
ส่งกลับขนาดพารามิเตอร์ที่โมเดล Cudnn RNN สามารถใช้ได้ การจัดสรรน้ำหนักและการเริ่มต้นครั้งต่อไปควรใช้ขนาดนี้
num_layers: ระบุจำนวนเลเยอร์ในโมเดล RNN num_units: ระบุขนาดของสถานะที่ซ่อนอยู่ input_size: ระบุขนาดของสถานะอินพุต 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 ของเมล็ดเพื่อเริ่มต้นการออกกลางคัน params_size: ขนาดของบัฟเฟอร์ params ที่ควรจัดสรรและเตรียมใช้งานสำหรับโมเดล RNN นี้ โปรดทราบว่าบัฟเฟอร์พารามิเตอร์นี้อาจเข้ากันไม่ได้กับ GPU โปรดใช้ CudnnRNNParamsWeights และ CudnnRNNParamsBiases เพื่อบันทึกและกู้คืนในลักษณะที่เข้ากันได้กับการทำงานที่แตกต่างกัน
คลาสที่ซ้อนกัน
ระดับ | CudnnRnnParamsSize.Options | แอ็ตทริบิวต์ทางเลือกสำหรับ CudnnRnnParamsSize |
ค่าคงที่
สตริง | OP_NAME | ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow |
วิธีการสาธารณะ
เอาท์พุต <U> | เป็นเอาท์พุต () ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์ |
คงที่ <U ขยาย TNumber , T ขยาย TNumber > CudnnRnnParamsSize <U> | สร้าง (ขอบเขต ขอบเขต ตัวดำเนินการ < TInt32 > numLayers, ตัวดำเนินการ < TInt32 > numUnits, ตัวดำเนินการ < TInt32 > inputSize, Class<T> T, Class<U> S, ตัวเลือก... ตัวเลือก) วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ CudnnRnnParamsSize ใหม่ |
CudnnRnnParamsSize.Options แบบคงที่ | ทิศทาง (ทิศทางของสตริง) |
CudnnRnnParamsSize.Options แบบคงที่ | การออกกลางคัน (การออกกลางคันแบบลอยตัว) |
CudnnRnnParamsSize.Options แบบคงที่ | inputMode (สตริงอินพุตโหมด) |
CudnnRnnParamsSize.Options แบบคงที่ | numProj (ยาว numProj) |
เอาท์พุต <U> | |
CudnnRnnParamsSize.Options แบบคงที่ | rnnMode (สตริง rnnMode) |
CudnnRnnParamsSize.Options แบบคงที่ | เมล็ด (เมล็ดยาว) |
CudnnRnnParamsSize.Options แบบคงที่ | เมล็ด2 (เมล็ดยาว2) |
วิธีการสืบทอด
ค่าคงที่
สตริงสุดท้ายแบบคงที่สาธารณะ OP_NAME
ชื่อของ op นี้ ซึ่งรู้จักกันในชื่อของเอ็นจิ้นหลัก TensorFlow
วิธีการสาธารณะ
เอาท์ พุท สาธารณะ <U> asOutput ()
ส่งกลับค่าแฮนเดิลสัญลักษณ์ของเทนเซอร์
อินพุตสำหรับการดำเนินการ TensorFlow คือเอาต์พุตของการดำเนินการ TensorFlow อื่น วิธีการนี้ใช้เพื่อรับหมายเลขอ้างอิงสัญลักษณ์ที่แสดงถึงการคำนวณอินพุต
สาธารณะ CudnnRnnParamsSize <U> สร้าง แบบคงที่ (ขอบเขต ขอบเขต , ตัวดำเนินการ < TInt32 > numLayers, ตัวดำเนินการ < TInt32 > numUnits, ตัวดำเนินการ < TInt32 > inputSize, Class<T> T, Class<U> S, ตัวเลือก... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่รวมการดำเนินการ CudnnRnnParamsSize ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ตัวเลือก | มีค่าแอตทริบิวต์ทางเลือก |
การส่งคืน
- อินสแตนซ์ใหม่ของ CudnnRnnParamsSize