bugfix: RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across

Update README.md

update readme

Update README.md

update readme

first commit

initial commit