Trả lời cho sự kiện TensorFlow Everywhere tại địa phương của bạn ngay hôm nay!
Trang này được dịch bởi Cloud Translation API.
Switch to English

dòng chảy :: ops :: 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 inputdiagonal , phép toán này trả về một tensor có hình dạng và giá trị giống như input , ngoại trừ đường chéo chính của ma trận trong cùng. Chúng sẽ bị ghi đè bởi các giá trị diagonal .

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

Giả sử inputk+1 kích thước [I, J, K, ..., M, N]diagonalk 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 cộng

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 cộng

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