Блог

Главная - Вопросы по программированию - получить форму тензора в java (я использую библиотеку org.pytorch)

получить форму тензора в java (я использую библиотеку org.pytorch)

#java #pytorch

Вопрос:

В python форма тензора (в PyTorch и Python) может быть получена с помощью

 print(tensor.shape)

Теперь я использую Android Studio (Java), и у меня есть примерный код, и тензор получается следующим образом

     import org.pytorch.Tensor;
    Tensor inputTensor = Tensor.fromBlob(inTensorBuffer, new long[]{1, 3, Constants.COUNT_OF_FRAMES_PER_INFERENCE, 160, 160});

Однако, когда я пересылаю ввод в мобильную модель PyTorch, ошибка исчезает, и я подозреваю, что размер ввода неверен для моей модели искусственного интеллекта. Поэтому я хочу получить размер(форму) Tensor inputTensor примерно (1,192,3,960,540)

Мое предположение состоит в том, что тензор равен 5D тензору, а первое измерение равно 1(пакет), а другие 4ea затемняются (количество кадров, RGB, ширина, высота). Я новичок в Java, в Java нет метода формы?

Метки: