rime.util.score_array.ElementWiseExpression

class rime.util.score_array.ElementWiseExpression(op, children)[source]

Tree representation of an element-wise expression; auto-broadcast

Methods

`__init__`(op, children)
`apply`(op)
`as_tensor`([device])
`collate_fn`(batch)	collate by rows for pytorch dataloaders
`eval`()
`exp`()
`numpy`()
`sigmoid`()
`softplus`()
`train`()
`traverse`([op_func])
`vae_module`(module)

Attributes

static collate_fn(batch)[source]: collate by rows for pytorch dataloaders