[英]Tensorflow C API: How to modify the value in tensor
如何在不創建新張量的情況下更改張量中的值?
例如,我有一個tensor
:
const size_t input_dims_len = 3;
int64_t input_dims[input_dims_len] = {1, 5, 12};
float input[5 * 12] = {...};
TF_Tensor* input_tensor = TF_NewTensor(TF_FLOAT,
input_dims,
input_dims_len,
input,
5 * 12 * sizeof(float),
deallocate_tensor,
nullptr);
當我更改input
的值時,如何使它也可以更改input_tensor
中的值?
// Allocate tensor
const int input_dims_len = 3;
int64_t input_dims[input_dims_len] = {1, 5, 12};
const int data_size = 5 * 12 * sizeof(float));
TF_Tensor* tensor = TF_AllocateTensor(TF_FLOAT,
input_dims, input_dims_len,
data_size);
// Change values in a tensor
float input[5 * 12] = {...};
std::memcpy(TF_TensorData(tensor), input, std::min(data_size, TF_TensorByteSize(tensor)));
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.