เทนเซอร์โฟลว์:: ปฏิบัติการ:: กระจัดกระจายSplit
#include <sparse_ops.h>
แยก SparseTensor
ออกเป็นเทนเซอร์ num_split
ตามมิติเดียว
สรุป
หาก shape[split_dim]
ไม่ใช่ผลคูณจำนวนเต็มของ num_split
ชิ้น [0 : shape[split_dim] % num_split]
ได้รับมิติพิเศษหนึ่งมิติ ตัวอย่างเช่น ถ้า split_dim = 1
และ num_split = 2
และอินพุตเป็น
input_tensor = shape = [2, 7] [ a d e ] [b c ]
กราฟิกเทนเซอร์เอาท์พุตคือ:
output_tensor[0] = shape = [2, 4] [ a ] [b c ] output_tensor[1] = shape = [2, 3] [ d e ] [ ]
ข้อโต้แย้ง:
- ขอบเขต: วัตถุ ขอบเขต
- split_dim: 0-D มิติที่จะแยกออกไป ต้องอยู่ในช่วง
[0, rank(shape))
- ดัชนี: เทนเซอร์ 2 มิติแสดงถึงดัชนีของเทนเซอร์แบบเบาบาง
- ค่า: เทนเซอร์ 1-D แทนค่าของเทนเซอร์แบบกระจาย
- รูปร่าง: 1-D เทนเซอร์แสดงถึงรูปร่างของเทนเซอร์แบบกระจัดกระจาย ดัชนีเอาท์พุต: รายการเทนเซอร์ 1-D แสดงถึงดัชนีของเทนเซอร์กระจัดกระจายเอาท์พุต
- num_split: จำนวนวิธีที่จะแยก
ผลตอบแทน:
-
OutputList
output_indices -
OutputList
output_values: รายการเทนเซอร์ 1-D แสดงถึงค่าของเทนเซอร์กระจัดกระจายเอาท์พุต -
OutputList
output_shape: รายการเทนเซอร์ 1-D แสดงถึงรูปร่างของเทนเซอร์กระจัดกระจายเอาท์พุต
ตัวสร้างและผู้ทำลาย | |
---|---|
SparseSplit (const :: tensorflow::Scope & scope, :: tensorflow::Input split_dim, :: tensorflow::Input indices, :: tensorflow::Input values, :: tensorflow::Input shape, int64 num_split) |
คุณลักษณะสาธารณะ | |
---|---|
operation | |
output_indices | |
output_shape | |
output_values |
คุณลักษณะสาธารณะ
การดำเนินการ
Operation operation
เอาท์พุท_ดัชนี
::tensorflow::OutputList output_indices
เอาท์พุท_รูปร่าง
::tensorflow::OutputList output_shape
เอาท์พุท_ค่า
::tensorflow::OutputList output_values
งานสาธารณะ
กระจัดกระจายSplit
SparseSplit( const ::tensorflow::Scope & scope, ::tensorflow::Input split_dim, ::tensorflow::Input indices, ::tensorflow::Input values, ::tensorflow::Input shape, int64 num_split )