Swap from torch.mm to torch.matmul to allow affine_field with offset#5

macrintr wants to merge 1 commit intomasterseung-lab/torchfields:masterfrom

tmacrina-affine-offsetseung-lab/torchfields:tmacrina-affine-offsetCopy head branch name to clipboard