Does torch.nn.batch_norm have support for sparse tensor? In my test:
torch.nn.batch_norm
Both ‘SparseCPU’ 和 ‘SparseCUDA’ backend doesn’t support.