Function image_tensor 

pub fn image_tensor(
    batch: DimExpr,
    channels: DimExpr,
    height: DimExpr,
    width: DimExpr,
) -> DependentType

4D image tensor (batch, channels, height, width).