tensorflow::ops::Stack

#include <array_ops.h>

Đóng gói một danh sách N tensor hạng R thành một tensor hạng (R+1) .

Bản tóm tắt

Đóng gói N tensor trong values ​​thành một tensor có hạng cao hơn một so với mỗi tensor trong values , bằng cách đóng gói chúng dọc theo chiều axis . Cho một danh sách các tensor có hình dạng (A, B, C) ;

Nếu axis == 0 thì tensor output sẽ có dạng (N, A, B, C) . Nếu axis == 1 thì tensor output sẽ có dạng (A, N, B, C) . Vân vân.

Ví dụ:

# 'x' is [1, 4]
# 'y' is [2, 5]
# 'z' is [3, 6]
pack([x, y, z]) => [[1, 4], [2, 5], [3, 6]]  # Pack along first dim.
pack([x, y, z], axis=1) => [[1, 2, 3], [4, 5, 6]]

Đây là hành động trái ngược với unpack .

Luận điểm:

  • phạm vi: Một đối tượng Phạm vi
  • Giá trị: Phải có cùng hình dạng và kiểu dữ liệu.

Các thuộc tính tùy chọn (xem Attrs ):

  • trục: Chiều mà theo đó để đóng gói. Giá trị âm sẽ được làm tròn, vì vậy phạm vi hợp lệ là [-(R+1), R+1) .

Trả về:

  • Output : Tensor đã được đóng gói.

Hàm khởi tạo và hàm hủy

Stack (const :: tensorflow::Scope & scope, :: tensorflow::InputList values)
Stack (const :: tensorflow::Scope & scope, :: tensorflow::InputList values, const Stack::Attrs & attrs)

Thuộc tính công khai

operation
output

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

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Các hàm tĩnh công khai

Axis (int64 x)

Cấu trúc

tensorflow::ops::Stack::Attrs

Các thiết lập thuộc tính tùy chọn cho Stack .

Thuộc tính công khai

hoạt động

Operation operation

đầu ra

::tensorflow::Output output

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

Ngăn xếp

 Stack(
  const ::tensorflow::Scope & scope,
  ::tensorflow::InputList values
)

Ngăn xếp

 Stack(
  const ::tensorflow::Scope & scope,
  ::tensorflow::InputList values,
  const Stack::Attrs & attrs
)

nút

::tensorflow::Node * node() const 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const 

Các hàm tĩnh công khai

Trục

Attrs Axis(
  int64 x
)