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

#include <array_ops.h>

Đảm bảo rằng hình dạng của tensor phù hợp với hình dạng mong đợi.

Tóm lược

Tăng lỗi nếu hình dạng của tensor đầu vào không khớp với hình dạng được chỉ định. Trả về giá trị tensor đầu vào nếu không.

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • input: Một tensor, có hình dạng cần được xác nhận.
  • shape: Hình dạng mong đợi (có thể được chỉ định một phần) của tensor đầu vào.

Lợi nhuận:

  • Output : Một tenxơ có hình dạng và nội dung giống với tenxơ đầu vào hoặc giá trị.

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

EnsureShape (const :: tensorflow::Scope & scope, :: tensorflow::Input input, PartialTensorShape shape)

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

EnsureShape

 EnsureShape(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  PartialTensorShape shape
)

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