I have a I have a matrix [5.5].

I have to average and get the matrix [1,5].

How do I do that? [5,5].

I have to average and get the matrix [1,5].

How do I do that?

Hi,

You can specify the dimension you want to reduce over in the mean functions:

`A.mean(dim=0)`

.