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

#include <array_ops.h>

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

Tóm lược

Cho một tensor input , hoạt động này trả về một tensor có dữ liệu đệm giống như input với datatype type .

Nếu datatype đầu vào T là lớn hơn so với sản lượng datatype type sau đó thay đổi hình dạng từ [...] để [..., 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 bằng sizeof ( type ) / sizeof ( T ). Sau đó, hình dạng đi từ [..., sizeof ( type ) / sizeof ( T )] đến [...].

tf.bitcast () và tf.cast () hoạt động khác nhau khi loại thực được truyền dưới dạng loại phức tạp (ví dụ: tf.complex64 hoặc tf.complex128) khi 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 (lần gọi gần đây nhất): ... InvalidArgumentError: Không thể phát bitcast từ 1 đến 18 [Op: Bitcast ] Equity_cast = tf.cast (a, tf.complex128) print (Equal_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, 3] y = [0, 2, 3, 3]. bitcoin shape = (3,), dtype = float32) print (equal_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 ép kiểu cấp thấp, vì vậy các máy có các kết quả cuối khác nhau sẽ cho các kết quả khác nhau.

Tranh luận:

Lợi nhuận:

  • Output : Bộ căng đầu ra.

Người xây dựng và Người phá 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ử :: tensorflow :: Đầu vào

 operator::tensorflow::Input() const

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

 operator::tensorflow::Output() const