flow
flow copied to clipboard
The shape of x in class DenseGridGen(nn.Module):
Hi What's the shape of x in the DenseGridGen in the grids.py?