Broadcast Operations

Contents

Broadcast Operations#

Tensor Operators#

add(x, y[, dim_0, dim_1])

Elementwise addition: \(x + y\).

sub(x, y[, dim_0, dim_1])

Elementwise subtraction: \(x - y\).

mul(x, y[, dim_0, dim_1])

Elementwise multiply: \(x * y\).

div(x, y[, dim_0, dim_1])

Elementwise division: \(x / y\).

sum(x[, dim])

Sum of array elements along with the corresponding dim.

min(x[, dim])

Min value of the array elements along with the corresponding dim.

max(x[, dim])

Max value of the array elements along with the corresponding dim.

argmin(x[, dim])

Return the indices of the minimum of an array along the dim.

argmax(x[, dim])

Return the indices of the maximum of an array along the dim.

mean(x[, dim])

Mean value of the array elements along with the corresponding dim.

std(x[, dim])

The standard deviation of the array elements along with the corresponding dim.

var(x[, dim])

The variance of the array elements along with the corresponding dim.

T(x[, dim])

dot(x, y[, dim_0, dim_1])

inv(x)

det(x)

diagonal_sum(x)