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.
Lập 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. - batch_variance
Output
: 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) | |
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 )
Đang đào tạo
Attrs IsTraining( bool x )