Google I / O hoạt động trở lại từ ngày 18 đến 20 tháng 5! Đặt chỗ và xây dựng lịch trình của bạn Đăng ký ngay

dòng chảy :: hoạt động :: Cây rơm

#include <array_ops.h>

Đóng gói một danh sách các tensor bậc N R thành một tensor bậc- (R+1) .

Tóm lược

Đóng gói N tensor trong values thành một tensor có bậc cao hơn một tensor trong values , bằng cách đóng gói chúng dọc theo kích thước axis . Cho một danh sách các tenxơ có dạng (A, B, C) ;

nếu axis == 0 thì tensor output sẽ có hình dạng (N, A, B, C) . nếu axis == 1 thì tensor output sẽ có hình 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]]

Điều này ngược lại với việc unpack .

Tranh luận:

  • 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.

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

  • axis: Kích thước cần đóng gói. Các giá trị âm bao quanh, vì vậy phạm vi hợp lệ là [-(R+1), R+1) .

Lợi nhuận:

  • Output : tensor đóng gói.

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

Chức năng tĩnh công khai

Axis (int64 x)

Cấu trúc

tensorflow :: ops :: Stack :: Attrs

Bộ thiết lập thuộc tính tùy chọn cho Ngăn xếp .

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

Cây rơm

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

Cây rơm

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

nút

::tensorflow::Node * node() const 

toán tử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const 

toán tử :: tensorflow :: Đầu ra

 operator::tensorflow::Output() const 

Chức năng tĩnh công khai

Trục

Attrs Axis(
  int64 x
)