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

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.

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