Method meanAxis
static member meanAxis: int -> Tensor<'T> -> Tensor<'T>
Calculates the mean of the elements along the specified axis.
Declaration
static member meanAxis: axis:int -> a:Tensor<'T> -> Tensor<'T>Parameters
| Type | Name | Description |
|---|---|---|
| int | axis | |
| Tensor<'T> | a | The tensor containing the source values. |
Returns
| Type | Description |
|---|---|
| Tensor<'T> | A new tensor containing the result of this operation. |
Remarks
The mean is calculated along the specified axis.
Examples
let a = HostTensor.ofList2D [[1.0; 2.0; 3.0; 4.0]
[5.0; 6.0; 7.0; 8.0]]
let b = Tensor.meanAxis 1 a // b = [2.5; 6.5]