ดึงข้อมูลพารามิเตอร์ CudnnRNN ในรูปแบบบัญญัติ รองรับการฉายภาพใน LSTM
ดึงชุดน้ำหนักจากบัฟเฟอร์ params แบบทึบที่สามารถบันทึกและกู้คืนได้ในลักษณะที่เข้ากันได้กับการทำงานในอนาคต
โปรดทราบว่าบัฟเฟอร์ params อาจเข้ากันไม่ได้ใน 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 dropout: ความน่าจะเป็นของการออกกลางคัน เมื่อตั้งค่าเป็น 0 การออกกลางคันจะถูกปิดใช้งาน เมล็ด: ส่วนที่ 1 ของเมล็ดเพื่อเริ่มต้นการออกกลางคัน seed2: ส่วนที่ 2 ของเมล็ดเพื่อเริ่มต้นการออกกลางคัน num_proj: มิติข้อมูลเอาต์พุตสำหรับเมทริกซ์การฉายภาพ ถ้าไม่มีหรือ 0 จะไม่มีการฉายภาพ
คลาสที่ซ้อนกัน
ระดับ | CudnnRNNParamsToCanonicalV2.Options | คุณลักษณะที่จำเป็นสำหรับการ CudnnRNNParamsToCanonicalV2 |
วิธีการสาธารณะ
รายการ < เอาท์พุท <T >> | อคติ () |
คง <T ขยายจำนวน> CudnnRNNParamsToCanonicalV2 <T> | |
คง CudnnRNNParamsToCanonicalV2.Options | ทิศทาง (ทิศทาง String) |
คง CudnnRNNParamsToCanonicalV2.Options | การออกกลางคัน (ลอยออกกลางคัน) |
คง CudnnRNNParamsToCanonicalV2.Options | inputMode (String inputMode) |
คง CudnnRNNParamsToCanonicalV2.Options | numProj (Long numProj) |
คง CudnnRNNParamsToCanonicalV2.Options | rnnMode (String rnnMode) |
คง CudnnRNNParamsToCanonicalV2.Options | เมล็ด (เมล็ดยาว) |
คง CudnnRNNParamsToCanonicalV2.Options | seed2 (Long seed2) |
รายการ < เอาท์พุท <T >> | น้ำหนัก () |
วิธีการสืบทอด
วิธีการสาธารณะ
สาธารณะคง CudnnRNNParamsToCanonicalV2 <T> สร้าง ( ขอบเขต ขอบเขต Operand <Integer> numLayers, Operand <Integer> numUnits, Operand <Integer> inputSize, Operand <T> params, numParamsWeights ยาวยาว numParamsBiases, ตัวเลือก ... ตัวเลือก)
วิธีการจากโรงงานเพื่อสร้างคลาสที่ปิดการดำเนินการ CudnnRNNParamsToCanonicalV2 ใหม่
พารามิเตอร์
ขอบเขต | ขอบเขตปัจจุบัน |
---|---|
ตัวเลือก | ดำเนินการค่าแอตทริบิวต์ทางเลือก |
คืนสินค้า
- อินสแตนซ์ใหม่ของ CudnnRNNParamsToCanonicalV2