encode method

Tensor encode(
  1. Tensor imageTensor
)

Run the encoder on an image tensor.

imageTensor: (batch, 3, height, width) — preprocessed image Returns: (batch, numPatches, encoderOutputDim)

Implementation

Tensor encode(Tensor imageTensor) {
  return encoder.forward(imageTensor);
}