Write a code on Python for simple neural network which can generate fixed third-dimensional arrays with size 64x64x64 and then train it on third-dimensional arrays of random numbers in range 0-1024. Check if CUDA available, if it is use it instead CPU