Fix iota dims check
Fix iota dims check

hshindo/arrayfire,pentschev/arrayfire,marbre/arrayfire,arrayfire/arrayfire,9prady9/arrayfire,munnybearz/arrayfire,ghisvail/arrayfire,umar456/arrayfire,umar456/arrayfire,bkloppenborg/arrayfire,merlin-ext/arrayfire,merlin-ext/arrayfire,victorv/arrayfire,pentschev/arrayfire,hshindo/arrayfire,bkloppenborg/arrayfire,arrayfire/arrayfire,shehzan10/arrayfire,munnybearz/arrayfire,umar456/arrayfire,pentschev/arrayfire,shehzan10/arrayfire,marbre/arrayfire,umar456/arrayfire,merlin-ext/arrayfire,arrayfire/arrayfire,arrayfire/arrayfire,shehzan10/arrayfire,marbre/arrayfire,hshindo/arrayfire,munnybearz/arrayfire,merlin-ext/arrayfire,pentschev/arrayfire,9prady9/arrayfire,bkloppenborg/arrayfire,9prady9/arrayfire,hshindo/arrayfire,victorv/arrayfire,victorv/arrayfire,ghisvail/arrayfire,shehzan10/arrayfire,bkloppenborg/arrayfire,ghisvail/arrayfire,9prady9/arrayfire,victorv/arrayfire,marbre/arrayfire,ghisvail/arrayfire