I have (pytorch) tensor object in java. (https://pytorch.org/javadoc/org/pytorch/Tensor.html) The shape of tensor is (3 x H x W) where 3 represent (R,G,B) each.
The