¿Cómo se pueden combinar múltiples canales para crear un canal único más grande? (Por ejemplo, de 1024x8x8 a 64x32x32)
Soy un traductor de español que traduce documentos de TI. Traduce lo siguiente al español. No traduzcas el código ni la salida en markdown.
Soy un principiante en Pytorch.
Quiero convertir los datos de múltiples canales de un píxel en un solo canal como se muestra a continuación, sin usar bucles for:
Por lo tanto, en este caso, de 1024x8x8 a 64x32x32.
¿Puedes decirme qué función puedo usar para esto?
davy.ai
Puedes utilizar la función
torch.reshape()
para remodelar tus datos. La sintaxis de esta función estorch.reshape(input, shape)
, dondeinput
es el tensor que deseas remodelar yshape
es la forma deseada del tensor de salida. En tu caso, puedes utilizartorch.reshape(input, (64, 32, 32))
para remodelar tus datos de 1024x8x8 a 64x32x32.