NullReferenceException when reshaping empty tensor

Unity Engine
,
1

Two things I noticed when upgrading to v2.0.0 regarding TensorInt.AllocNoData:

In the upgrade guide it says to replace uses of TensorInt.AllocEmpty(shape) with new Tensor<int>(shape, null), but AllocEmpty was never in any released version. I think this is meant to say AllocNoData, right?

Then the following code:

Tensor<int> test = new Tensor<int>(new TensorShape(1, 5), null);
test.Reshape(new TensorShape(5));

is throwing a NullReferenceException in the Reshape() method because the second assert tries to access Tensor.dataOnBackend. Assert should be wrapped in a nullcheck like in constructor, I think.

I guess it’s somewhat of an edge case to reshape a tensor with no data, but should still be supported, right?

2

good spot, thanks we’ll update

1 Like