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 https://github.com/cuishuhao/BNM/issues/3 might work. We also speed up BNM in recent unpublished work, which might be released in September.

[mzprose]

OK，thanks for your reply！