VortexNet: Neural network based on fluid dynamics