How to repeat tensor in libtorch

Question

I am using libtorch to inference, I have read data from txt file to vector and convert to tensor, I want to repeat a tensor three times then change it to 3D, I tried this

std::vector<std::vector<float>> feature_data(255, std::vector<float>(221));
ifstream f_data("../data.txt"); // 
if (! f_data) {
    cout << "Error, file couldn't be opened" << endl; 
    return 1; 
}    
for(int i=0;i<255;i++)
{
    for(int j=0;j<221;j++)
    {
        if ( !f_data ) 
        {
            std::cout << "read error" << std::endl;
            break;
        }
        f_data >> feature_data[i][j];
    }
}
auto data_options = torch::TensorOptions().dtype(at::kFloat);
auto feature_tensor = torch::zeros({255,221}, data_options);
for (int i = 0; i < 255; i++)
    feature_tensor.slice(0, i,i+1) = torch::from_blob(feature_data[i].data(), {221}, 
    data_options);

// begin to repeat three times
auto tensor_clone = feature_tensor.clone();
auto one_time_clone = torch::cat({feature_tensor, tensor_clone}, 0);
auto two_times_clone = torch::cat({one_time_clone, tensor_clone}, 0);
auto transformed_asr = two_times_clone.view({3, 255, 221});

it looks troublesome and I am not sure if it is right, is there an easy way?

Yongqi Z · Accepted Answer · 2022-03-02 10:36:42Z

0

...
for (int i = 0; i < 255; i++)
    feature_tensor.slice(0, i,i+1) = torch::from_blob(feature_data[i].data(), {221}, 
data_options);
auto tensor_clone = feature_tensor.repeat({3, 1});
...

answered Mar 2, 2022 at 10:36

Yongqi Z

66110 silver badges24 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to repeat tensor in libtorch

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related