tensorflow::ops::IdentityN

#include <array_ops.h>

Trả về một danh sách các tensor có cùng hình dạng và nội dung với tensor đầu vào.

Bản tóm tắt

tenxơ.

Thao tác này có thể được sử dụng để ghi đè đạo hàm cho các hàm phức tạp. Ví dụ, giả sử y = f(x) và chúng ta muốn áp dụng một hàm tùy chỉnh g cho quá trình lan truyền ngược sao cho dx = g(dy). Trong Python,

with tf.get_default_graph().gradient_override_map(
    {'IdentityN': 'OverrideGradientWithG'}):
  y, _ = identity_n([f(x), x])

.RegisterGradient('OverrideGradientWithG')
def ApplyG(op, dy, _):
  return [None, g(dy)]  # Do not backprop to f(x).

Luận điểm:

Trả về:

  • OutputList : Tensor đầu ra.

Hàm khởi tạo và hàm hủy

IdentityN (const :: tensorflow::Scope & scope, :: tensorflow::InputList input)

Thuộc tính công khai

operation
output

Chức năng công cộng

operator[] (size_t index) const

Thuộc tính công khai

hoạt động

Operation operation

đầu ra

::tensorflow::OutputList output

Chức năng công cộng

IdentityN

 IdentityN(
  const ::tensorflow::Scope & scope,
  ::tensorflow::InputList input
)

toán tử[]

::tensorflow::Output operator[](
  size_t index
) const