Module ops

Module ops 

Source
Expand description

operator like traits for common tensor operations

Structs§

UnsqueezeScalar
unsqueezed scalar that can be converted to vector type

Traits§

Abs
trait to represent the operation
Cat
trait to represent the operation
Flatten
Rank
get tensor rank
Reshape
SquaredError
trait to represent the operation
Squeeze
trait to represent the operation
Stack
trait to represent the operation
SwapDims
trait to represent the operation
Unsqueeze
trait to represent the operation