Specify the shape only at the allocation stage