dòng chảy căng:: ôi:: Bitcast

#include <array_ops.h>

Bitcast một tensor từ loại này sang loại khác mà không cần sao chép dữ liệu.

Bản tóm tắt

Với một input tensor, thao tác này trả về một tensor có cùng dữ liệu đệm với input có type dữ liệu.

Nếu kiểu dữ liệu đầu vào T lớn hơn type dữ liệu đầu ra thì hình dạng sẽ thay đổi từ [...] thành [..., sizeof( T )/sizeof( type )].

Nếu T nhỏ hơn type , toán tử yêu cầu kích thước ngoài cùng bên phải phải bằng sizeof( type )/sizeof( T ). Sau đó, hình dạng sẽ đi từ [..., sizeof( type )/sizeof( T )] đến [...].

tf.bitcast() và tf.cast() hoạt động khác nhau khi dtype thực được tạo thành một dtype phức tạp (ví dụ: tf.complex64 hoặc tf.complex128) vì tf.cast() tạo phần ảo 0 trong khi tf.bitcast() cung cấp mô-đun lỗi. Ví dụ,

Ví dụ 1:

a = [1., 2., 3.] Equal_bitcast = tf.bitcast(a, tf.complex128) Traceback (cuộc gọi gần đây nhất gần đây nhất): ... InvalidArgumentError: Không thể bitcast từ 1 đến 18 [Op: Bitcast ] Equality_cast = tf.cast(a, tf.complex128) print(equality_cast) tf.Tensor([1.+0.j 2.+0.j 3.+0.j], shape=(3,), dtype=complex128)

Ví dụ 2:

tf.bitcast(tf.constant(0xffffffff, dtype=tf.uint32), tf.uint8)

Ví dụ 3:

x = [1., 2., 3.] y = [0., 2., 3.] đẳng thức= tf.equal(x,y) đẳng_cast = tf.cast(equality,tf.float32) đẳng_bitcast = tf. bitcast(equality_cast,tf.uint8) print(equality) tf.Tensor([False True True], shape=(3,), dtype=bool) print(equality_cast) tf.Tensor([0. 1. 1.], hình dạng=(3,), dtype=float32) print(equality_bitcast) tf.Tensor( [[ 0 0 0 0] [ 0 0 128 63] [ 0 0 128 63]], shape=(3, 4), dtype= uint8)

LƯU Ý : Bitcast được triển khai dưới dạng truyền cấp độ thấp, vì vậy các máy có thứ tự cuối khác nhau sẽ cho kết quả khác nhau.

Lập luận:

Trả về:

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

Bitcast (const :: tensorflow::Scope & scope, :: tensorflow::Input input, DataType type)

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

Bitcast 

 Bitcast(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  DataType type
)

nút 

::tensorflow::Node * node() const

toán tử::tenorflow::Đầu vào 

 operator::tensorflow::Input() const

toán tử::tenorflow::Đầu ra

 operator::tensorflow::Output() const