I’m trying to get a reshape layer to work, I’m first creating a shape like this
let batch_size = 10 let shape:Tensor<Int32> = [batch_size, 7, 7, 256]
But then I get this error
error: <Cell 57>:2:28: error: cannot convert value of type 'Int' to expected element type '_TensorElementLiteral<Int32>' let shape:Tensor<Int32> = [batch_size, 7, 7, 256] ^~~~~~~~~~ _TensorElementLiteral<Int32>( )
What is the hack here to get this to work?