Python shape_of примеры использования

Пример #1

 def size(value):
     return (op.prod(op.shape_of(value)), )

Пример #2

 def size(unused_ctx, value):
     return (op.cast(op.prod(op.shape_of(value)), plaidml.DType.INT64), )

Пример #3

 def shape(data):
     return (op.shape_of(data), )

Пример #4

 def shape(unused_ctx, data):
     return (op.cast(op.shape_of(data), plaidml.DType.INT64), )