tensorflow::ops::Bitcast

#include <array_ops.h>

Chuyển đổi kiểu dữ liệu tensor từ kiểu này sang kiểu khác mà không cần sao chép dữ liệu.

Bản tóm tắt

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

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

Nếu T nhỏ hơn type , toán tử yêu cầu chiều ngoài cùng bên phải phải bằng kích thước của type / kích thước của T Hình dạng sau đó chuyển từ [..., kích thước của type / kích thước của T ] sang [...].

tf.bitcast() và tf.cast() hoạt động khác nhau khi kiểu dữ liệu thực được ép kiểu thành kiểu dữ liệu phức (ví dụ: tf.complex64 hoặc tf.complex128) vì tf.cast() đặt phần ảo bằng 0 trong khi tf.bitcast() gây ra lỗi mô-đun. Ví dụ:

Ví dụ 1:

a = [1., 2., 3.] equality_bitcast = tf.bitcast(a, tf.complex128) Traceback (most recent call last): ... InvalidArgumentError: Cannot bitcast from 1 to 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.] equality= tf.equal(x,y) equality_cast = tf.cast(equality,tf.float32) equality_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.], shape=(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 phép ép kiểu cấp thấp, do đó các máy có thứ tự byte khác nhau sẽ cho kết quả khác nhau.

Luận điểm:

Trả về:

Hàm khởi 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 

operator::tensorflow::Input

 operator::tensorflow::Input() const 

operator::tensorflow::Output

 operator::tensorflow::Output() const