ML Topluluk Günü 9 Kasım! TensorFlow, JAX güncellemeler için bize katılın ve daha fazla bilgi edinin

tensorflow :: ops :: DiagPart

#include <array_ops.h>

Tensörün köşegen kısmını döndürür.

Özet

Bu işlem, input diagonal kısmı ile bir tensör döndürür. diagonal kısım şu şekilde hesaplanır:

input [D1,..., Dk, D1,..., Dk] boyutlarına sahip olduğunu varsayın, ardından çıktı, [D1,..., Dk] boyutlarına sahip k derecesinin bir tensörüdür, burada:

diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik] .

Örneğin:

# 'input' is [[1, 0, 0, 0]
              [0, 2, 0, 0]
              [0, 0, 3, 0]
              [0, 0, 0, 4]]

tf.diag_part(input) ==> [1, 2, 3, 4]

Argümanlar:

  • kapsam: Bir Scope nesnesi
  • girdi: Derece k tensörü burada k eşittir ve sıfır değildir.

İadeler:

  • Output : Çıkarılan köşegen.

Yapıcılar ve Yıkıcılar

DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input)

Genel özellikler

diagonal
operation

Kamusal işlevler

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Genel özellikler

diyagonal

::tensorflow::Output diagonal

operasyon

Operation operation

Kamusal işlevler

DiagPart

 DiagPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input
)

düğüm

::tensorflow::Node * node() const 

operator :: tensorflow :: Girdi

 operator::tensorflow::Input() const 

operator :: tensorflow :: Çıktı

 operator::tensorflow::Output() const