Skip to contents

Shape of a lazy tensor. Might be NULL if the shapes is not known or varying between rows. Batch dimension is always NA.

Usage

lazy_shape(x)

Arguments

x

(lazy_tensor)
Lazy tensor.

Value

(integer() or NULL)

Examples

lt = as_lazy_tensor(1:10)
lazy_shape(lt)
#> [1] NA  1
lt = as_lazy_tensor(matrix(1:10, nrow = 2))
lazy_shape(lt)
#> [1] NA  5