dòng chảy căng:: ôi:: Phân chia thưa thớt

Chia một SparseTensor thành các tensor num_split dọc theo một chiều.

Bản tóm tắt

Nếu shape[split_dim] không phải là bội số nguyên của num_split . Lát [0 : shape[split_dim] % num_split] có thêm một chiều. Ví dụ: nếu split_dim = 1 và num_split = 2 và đầu vào là

input_tensor = shape = [2, 7]
[    a   d e  ]
[b c          ]

Về mặt đồ họa, các tensor đầu ra là:

output_tensor[0] = shape = [2, 4]
[    a  ]
[b c    ]

output_tensor[1] = shape = [2, 3]
[ d e  ]
[      ]

Lập luận:

• phạm vi: Một đối tượng Phạm vi
• chia_dim: 0-D. Kích thước để phân chia. Phải nằm trong phạm vi [0, rank(shape)) .
• chỉ số: tensor 2-D biểu thị các chỉ số của tensor thưa thớt.
• giá trị: tensor 1-D đại diện cho các giá trị của tensor thưa thớt.
• hình dạng: 1-D. tensor đại diện cho hình dạng của tensor thưa thớt. chỉ số đầu ra: Danh sách các tensor 1-D biểu thị chỉ số của tensor thưa thớt đầu ra.
• num_split: Số cách chia.

Trả về:

• OutputList
• OutputList đầu ra_values: Danh sách các tensor 1-D biểu thị các giá trị của tensor thưa thớt đầu ra.
• OutputList đầu ra_shape: Danh sách các tensor 1-D biểu thị hình dạng của tensor thưa thớt đầu ra.

Thuộc tính công khai

Operation operation

::tensorflow::OutputList output_indices

::tensorflow::OutputList output_shape

::tensorflow::OutputList output_values

 SparseSplit(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input split_dim,
  ::tensorflow::Input indices,
  ::tensorflow::Input values,
  ::tensorflow::Input shape,
  int64 num_split
)