Relative Content

Tag Archive for pythontensorflowmachine-learningneural-network

Single-headed transformer model, output and target tensor sizes don’t match

I’m trying to build a simple single-headed transformer, with # input -> embedding -> single-head transformer -> dense -> output these layers, I think I have done too much, and now I keep getting an error that says my input and output tensor sizes don’t match and I think my error is in my forward method because that’s what I see in my debugging statements