dòng chảy :: hoạt động :: SparseSplit

#include <sparse_ops.h>

Tách SparseTensor thành các tensors 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 . Slices [0 : shape[split_dim] % num_split] có thêm một kích thước. 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, bộ căng đầu ra là: 

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

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

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • split_dim: 0-D. Kích thước cùng với đó để phân chia. Phải nằm trong khoảng [0, rank(shape)) .
  • chỉ số: tensor 2-D đại diện cho các chỉ số của tensor thưa thớt.
  • giá trị: 1-D tensor đạ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: Một danh sách các tenxơ 1-D đại diện cho các chỉ số của các tenxơ đầu ra thưa thớt.
  • num_split: Số cách tách.

Lợi nhuận:

  • OutputList output_indices
  • OutputList output_values: Danh sách các tenxơ 1-D đại diện cho các giá trị của các tenxơ thưa thớt đầu ra.
  • OutputList output_shape: Danh sách các tenxơ 1-D đại diện cho hình dạng của các tenxơ thưa thớt đầu ra.

Người xây dựng và Người phá 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 cộng

operation
output_indices
output_shape
output_values

Thuộc tính công cộng

hoạt động

Operation operation

output_indices

::tensorflow::OutputList output_indices

output_shape

::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
)