Tensor

[mzprose]

Hello,if I have a Tensor with size [8,3,512,512], how can I compute BNM?

[cuishuhao]

That depends on the goal of optimization the tensor.
If batch_size=8, class_num=3, width=height=512,
you can resize the tensor to [8 * 512 * 512, 3], to ensure class_num (3) as a dimension.
Then BNM loss can be calculated on the tensor with size of [8 * 512 * 512, 3].

[mzprose]

Thank you for answering, I will try, but in this case, there is a question, is the matrix too large, I am afraid it is difficult to execute the program.

[cuishuhao]

Tricks in #3 might work.
We also speed up BNM in recent unpublished work, which might be released in September.

[mzprose]

OK，thanks for your reply！