dòng chảy căng:: ôi:: Hợp nhấtBatchNorm

#include <nn_ops.h>

Chuẩn hóa hàng loạt.

Bản tóm tắt

Lưu ý rằng kích thước của Tensor 4D được xác định bởi "NHWC" hoặc "NCHW". Kích thước của Tensor 1D phù hợp với kích thước C của Tensor 4D.

Tranh luận:

  • phạm vi: Một đối tượng Phạm vi
  • x: Tensor 4D cho dữ liệu đầu vào.
  • tỷ lệ: Tenor 1D cho hệ số tỷ lệ, để chia tỷ lệ x chuẩn hóa.
  • offset: Tenor 1D để bù, để chuyển sang x chuẩn hóa.
  • có nghĩa là: Một Tenor 1D cho giá trị trung bình của dân số. Chỉ được sử dụng để suy luận; phải trống để đào tạo.
  • phương sai: Tenor 1D cho phương sai tổng thể. Chỉ được sử dụng để suy luận; phải trống để đào tạo.

Thuộc tính tùy chọn (xem Attrs ):

  • epsilon: Một số float nhỏ được thêm vào phương sai của x.
  • data_format: Định dạng dữ liệu cho x và y. "NHWC" (mặc định) hoặc "NCHW".
  • is_training: Giá trị bool để biểu thị thao tác là dành cho đào tạo (mặc định) hoặc suy luận.

Trả về:

  • Output y: Tensor 4D cho dữ liệu đầu ra.
  • Lô_mean Output : Tenor 1D cho giá trị trung bình lô được tính toán, được TensorFlow sử dụng để tính giá trị trung bình đang chạy.
  • Output batch_variance: Tensor 1D cho phương sai lô được tính toán, được TensorFlow sử dụng để tính toán phương sai đang chạy.
  • Dự trữ_space_1 Output : Tenor 1D cho giá trị trung bình lô được tính toán, sẽ được sử dụng lại trong tính toán độ dốc.
  • Dự trữ_space_2 Output : Tenor 1D cho phương sai lô được tính toán (phương sai đảo ngược trong trường hợp cuDNN), được sử dụng lại trong tính toán độ dốc.

Hàm tạo và hàm hủy

FusedBatchNorm (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance)
FusedBatchNorm (const :: tensorflow::Scope & scope, :: tensorflow::Input x, :: tensorflow::Input scale, :: tensorflow::Input offset, :: tensorflow::Input mean, :: tensorflow::Input variance, const FusedBatchNorm::Attrs & attrs)

Thuộc tính công khai

batch_mean
batch_variance
operation
reserve_space_1
reserve_space_2
y

Các hàm tĩnh công khai

DataFormat (StringPiece x)
Epsilon (float x)
ExponentialAvgFactor (float x)
IsTraining (bool x)

Cấu trúc

tensorflow:: ops:: FusedBatchNorm:: Attrs

Trình thiết lập thuộc tính tùy chọn cho FusedBatchNorm .

Thuộc tính công khai

batch_mean

::tensorflow::Output batch_mean

lô_phương sai

::tensorflow::Output batch_variance

hoạt động

Operation operation

dự trữ_space_1

::tensorflow::Output reserve_space_1

dự trữ_space_2

::tensorflow::Output reserve_space_2

y

::tensorflow::Output y

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

Hợp nhấtBatchNorm

 FusedBatchNorm(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance
)

Hợp nhấtBatchNorm

 FusedBatchNorm(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input x,
  ::tensorflow::Input scale,
  ::tensorflow::Input offset,
  ::tensorflow::Input mean,
  ::tensorflow::Input variance,
  const FusedBatchNorm::Attrs & attrs
)

Các hàm tĩnh công khai

Định dạng dữ liệu

Attrs DataFormat(
  StringPiece x
)

Epsilon

Attrs Epsilon(
  float x
)

Hệ số trung bình lũy thừa

Attrs ExponentialAvgFactor(
  float x
)

Đó là hành trình

Attrs IsTraining(
  bool x
)