How to create a random tensor with a given shape in pytorch lite (Java)?

Question

I want to create a random tensor with a shape (BxHxWxC) to test my ptl model. How can I create it?

I see fromBlob but not any method to create a random tensor with a specific shape which is pretty normal in pytorch to test a model.

user_4622046 · Answer

This one works for initializing a tensor with all zeros.

FloatBuffer buffer = Tensor.allocateFloatBuffer(1*224*224*3);
Tensor tensor = Tensor.fromBlob(buffer, new long[]{1, 224, 224, 3});