Crate glowstick_candle Copy item path Source pub use tensor::Error ;pub use tensor::Tensor ;op tensor broadcast_add Performs addition of the lefthand tensor and righthand tensor(s). The righthand
tensor(s) must be compatible for broadcast to the shape of the lefthand tensor. cat Concatenates the given tensors along a specified dimension.
A dynamic dimension must be provided for the return type. conv2d Applies a 2D convolution over the input tensor with the provided kernel, padding,
dilation, stride and groups. expand Broadcasts the lefthand tensor to the shape of the provided righthand tensor
or shape. flatten Flattens the given tensor from the specified start dimension to the end
dimension. gather Gathers the elements from a tensor at the provided indices along a specified dimension. log_softmax Applies the log softmax function to a tensor along the specified dimension. matmul Performs matrix multiplication of the lefthand tensor and righthand tensor(s). narrow Narrows a tensor at the specified dimension from start index to length. reshape Reshapes a tensor to the specified dimensions. reshape_tys reshape_val softmax Applies the softmax function to a tensor along the specified dimension. squeeze Squeezes the specified dimensions from a tensor. squeeze_next transpose Swaps the dimensions of a tensor. unsqueeze Unsqueezes a tensor at the specified dimension(s).