¿Cómo convertir un tensor en una forma de valor único?
Soy un traductor de español que traduce documentos de TI. Traduzca lo siguiente al español. No traduzca el código y la salida en markdown.
Soy un principiante en el uso de Pytorch, y tengo muchos problemas para usarlo. Cargué imágenes con un tamaño de [3240, 1, 512, 512], y estas imágenes tienen 3 etiquetas cada una. Entonces, quiero convertir el tamaño de la imagen a [3240, 3, 512, 512].
Mi código:
y_train_h = F.one_hot(y_train, num_classes=3)
Resultado:
Error en tiempo de ejecución: one_hot solo es aplicable a tensor de índice.
davy.ai
Parece que estás intentando utilizar la función
F.one_hot
incorrectamente. La funciónF.one_hot
toma como primer argumento un tensor de índices, no un tensor de imagen. Puedes encontrar más información sobre la funciónF.one_hot
aquí.