tensorflow::ops::SparseSplit

#include <sparse_ops.h>

Chia một SparseTensor thành num_split tensor 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 , thì Slices [0 : shape[split_dim] % num_split] sẽ 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 như sau:

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

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

Luận điểm:

  • phạm vi: Một đối tượng Phạm vi
  • split_dim: 0-D. Chiều mà theo đó sẽ chia. Phải nằm trong khoảng [0, rank(shape)) .
  • chỉ số: Tensor 2 chiều biểu thị các chỉ số của tensor thưa.
  • values: Tensor 1 chiều biểu thị các giá trị của tensor thưa.
  • shape: Tensor 1 chiều biểu thị hình dạng của tensor thưa. output indices: Một danh sách các tensor 1 chiều biểu thị các chỉ số của các tensor thưa đầu ra.
  • num_split: Số cách chia.

Trả về:

  • OutputList chỉ số đầu ra
  • OutputList output_values: Một danh sách các tensor 1 chiều biểu thị các giá trị của các tensor thưa đầu ra.
  • OutputList output_shape: Một danh sách các tensor 1 chiều biểu thị hình dạng của các tensor thưa đầu ra.

Hàm khởi 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

chỉ số đầu ra 

::tensorflow::OutputList output_indices

hình dạng đầu ra 

::tensorflow::OutputList output_shape

giá trị đầu ra 

::tensorflow::OutputList output_values

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

SparseSplit 

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