def phi(A):
"""
Return lower triangle of A and halve the diagonal.
"""
B = A.tril()
B = B - 0.5 * torch.diag(torch.diag(B))
return B
微信
公众号
扫码关注公众号