now julia code fetches tensor sizes automatically