thanks, that looks to have fixed that bit. I think where i have a lack of knowledge is that i’m confused around what needs to be passed around in the tensor at which points. i’m thinking of batch size, channels, rows and columns. the ‘1’ here seemed  ok to me as either channel or row but in fact neither was needed !