dòng chảy :: hoạt động :: MatrixSetDiag

#include <array_ops.h>

Trả về tensor ma trận theo lô với các giá trị đường chéo theo lô mới.

Tóm lược

Đã cho input và diagonal , phép toán này trả về một tensor có cùng hình dạng và giá trị như input , ngoại trừ đường chéo chính của ma trận trong cùng. Các giá trị này sẽ bị ghi đè bởi các giá trị diagonal .

Kết quả đầu ra được tính như sau:

Giả sử input có k+1 kích thước [I, J, K, ..., M, N] và diagonal có k nguyên [I, J, K, ..., min(M, N)] . Khi đó đầu ra là một tenxơ hạng k+1 với các kích thước [I, J, K, ..., M, N] trong đó:

  • output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] cho m == n .
  • output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] cho m != n .

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • đầu vào: Xếp hạng k+1 , trong đó k >= 1 .
  • đường chéo: Xếp hạng k , trong đó k >= 1 .

Lợi nhuận:

  • Output : Xếp hạng k+1 , với output.shape = input.shape .

Người xây dựng và Người phá hủy

MatrixSetDiag (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input diagonal)

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

MatrixSetDiag

 MatrixSetDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input diagonal
)

nút

::tensorflow::Node * node() const

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

 operator::tensorflow::Input() const

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

 operator::tensorflow::Output() const