dòng chảy căng:: ôi:: Mở rộngDims

#include <array_ops.h>

Chèn kích thước 1 vào hình dạng của tenxơ.

Bản tóm tắt

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

Thao tác này rất hữu ích nếu bạn muốn thêm thứ nguyên lô vào một phần tử. Ví dụ: nếu bạn có một hình ảnh duy nhất có hình dạng [height, width, channels] , bạn có thể tạo thành một lô gồm 1 hình ảnh với expand_dims(image, 0) , điều này sẽ tạo thành hình dạng [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]

Hoạt động này yêu cầu:

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

Hoạt động này có liên quan đến squeeze() , loại bỏ các kích thước có kích thước 1.

Lập luận:

  • phạm vi: Một đối tượng phạm vi
  • trục: 0-D (vô hướng). Chỉ định chỉ số thứ nguyên để mở rộng hình dạng của input . Phải nằm trong phạm vi [-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ó kích thước bổ sung được thêm vào là 1.

Hàm tạo và hàm hủy

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

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

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

Mở rộngDims 

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

nút 

::tensorflow::Node * node() const

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

 operator::tensorflow::Input() const

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

 operator::tensorflow::Output() const