#!/usr/bin/env python3

import torch

aa = torch.arange(10.0).reshape(2,5)
bb = torch.arange(5) + 1.0
cc = torch.divide(aa,bb)
print("aa")
print(aa)
print("bb")
print(bb)
print("cc")
print(cc)
print()

print("# ======================")
print("# row-wise normalization")
print("# ======================")
aaSum   = aa.sum(1)                     # aaSum = aa.pow(2).sum(1) 
aaTrans = torch.divide(aa.transpose(0,1),aaSum)
aaNorm  = torch.transpose(aaTrans,0,1)
print("asSum")
print(aaSum)
print("asTrans")
print(aaTrans)
print("aaNorm")
print(aaNorm)