PyTorch: PyTorch में numpy.linalg.multi_dot () के बराबर क्या है?

मैं PyTorch में कई मैट्रिक्स के मैट्रिक्स गुणन करने की कोशिश कर रहा हूं और सोच रहा था कि PyTorch में numpy.linalg.multi_dot() के बराबर क्या है?

यदि कोई नहीं है, तो अगला सबसे अच्छा तरीका क्या है (गति और स्मृति के मामले में) मैं इसे PyTorch में कर सकता हूं?

कोड:

import numpy as np
import torch

A = np.random.rand(3, 3)
B = np.random.rand(3, 3)
C = np.random.rand(3, 3)

results = np.linalg.multi_dot(A, B, C)

A_tsr = torch.tensor(A)
B_tsr = torch.tensor(B)
C_tsr = torch.tensor(C)

# What is the PyTorch equivalent of np.linalg.multi_dot()?

बहुत धन्यवाद!