Changing tensor datatype

ND4J
1

I have a tensor with INT datatype that I need to convert into a tensor of floats.

I can’t find any method allowing me to do that in the documentation… am I missing something?

2

INDArray.castTo is what you are looking for.

1 Like