tensorflow:: ops:: ExpandDims

Wstawia wymiar 1 do kształtu tensora.

Streszczenie

Biorąc pod uwagę tensor input , ta operacja wstawia wymiar 1 na axis wymiaru o indeksie kształtu elementu input . axis wymiaru o indeksie zaczyna się od zera; jeśli podasz liczbę ujemną dla axis jest ona liczona wstecz od końca.

Ta operacja jest przydatna, jeśli chcesz dodać wymiar wsadowy do pojedynczego elementu. Na przykład, jeśli masz pojedynczy obraz o kształcie [height, width, channels] , możesz utworzyć wsad 1 obrazu za pomocą expand_dims(image, 0) , co spowoduje utworzenie kształtu o wymiarach [1, height, width, channels] .

Inne przykłady:

# '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]

Do przeprowadzenia tej operacji wymagane są następujące czynności:

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

Operacja ta jest powiązana z squeeze() , która usuwa wymiary o rozmiarze 1.

Argumenty:

• zakres: obiekt zakresu
• Oś: 0-D (skalar). Określa indeks wymiaru, do którego ma zostać rozszerzony kształt input . Musi mieścić się w zakresie [-rank(input) - 1, rank(input)] .

Zwroty:

• Output : Zawiera te same dane co input , ale jego kształt ma dodatkowy wymiar o rozmiarze 1.

Atrybuty publiczne

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