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

#include <sparse_ops.h>

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 = 1num_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.

Hàm tạo và hàm hủy

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

Thuộc tính công khai

operation
output_indices
output_shape
output_values

Thuộc tính công khai

hoạt động

Operation operation

đầu ra_indices

::tensorflow::OutputList output_indices

đầu ra_hình dạng

::tensorflow::OutputList output_shape

giá trị đầu ra

::tensorflow::OutputList output_values

Chức năng công cộng

Phân chia thưa thớt

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