tensorflow::ops::ExpandDims

Chèn thêm chiều 1 vào hình dạng của tensor.

Bản tóm tắt

Với một tensor input , thao tác này chèn một chiều có giá trị 1 vào axis chỉ số chiều của hình dạng input . axis chỉ số chiều bắt đầu từ số 0; nếu bạn chỉ định một số âm cho axis nó sẽ được đếm ngược từ cuối.

Thao tác này hữu ích nếu bạn muốn thêm chiều hàng loạt vào một phần tử đơn lẻ. Ví dụ, nếu bạn có một hình ảnh đơn lẻ có kích thước [height, width, channels] , bạn có thể biến nó thành một hàng loạt gồm 1 hình ảnh bằng cách sử dụng expand_dims(image, 0) , điều này sẽ tạo ra kích thước [1, height, width, channels] .

Các ví dụ khác:

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]

Thao tác này yêu cầu:

-1-input.dims() <= dim <= input.dims()

Thao tác này có liên quan đến squeeze() , hàm này loại bỏ các chiều có kích thước bằng 1.

Luận điểm:

• phạm vi: Một đối tượng Phạm vi
• trục: 0-D (vô hướng). Chỉ định chỉ số chiều mà tại đó hình dạng của input sẽ được mở rộng. Phải nằm trong khoảng [-rank(input) - 1, rank(input)] .

Trả về:

• Output : Chứa dữ liệu giống như input , nhưng hình dạng của nó có thêm một chiều kích thước là 1.

Thuộc tính công khai

Operation operation

::tensorflow::Output output

 ExpandDims(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input axis
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const